numpy.diag_indices_from #
- 麻木的。diag_indices_from ( arr ) [来源] #
返回用于访问 n 维数组主对角线的索引。
请参阅
diag_indices
了解完整详细信息。- 参数:
- arr数组,至少二维
也可以看看
笔记
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]))