numpy.diag_indices_from

numpy.diag_indices_from(arr)

返回用於存取 n 維陣列主對角線的索引。

完整細節請參閱 diag_indices。

參數：:

arr陣列，至少 2 維

另請參閱

diag_indices

範例

>>> import numpy as np

建立一個 4x4 陣列。

>>> a = np.arange(16).reshape(4, 4)
>>> a
array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11],
       [12, 13, 14, 15]])

取得對角線元素的索引。

>>> di = np.diag_indices_from(a)
>>> di
(array([0, 1, 2, 3]), array([0, 1, 2, 3]))

>>> a[di]
array([ 0,  5, 10, 15])

這只是 diag_indices 的語法糖。

>>> np.diag_indices(a.shape[0])
(array([0, 1, 2, 3]), array([0, 1, 2, 3]))