numpy.diag_indices_from #

麻木的。diag_indices_from ( arr ) [来源] #

返回用于访问 n 维数组主对角线的索引。

请参阅diag_indices了解完整详细信息。

参数
arr数组,至少二维

也可以看看

diag_indices

笔记

1.4.0 版本中的新增功能。

例子

创建一个 4 x 4 数组。

>>> 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]))