numpy.diag_indices_from#
- numpy.diag_indices_from(arr)[source]#
返回访问 n 维数组主对角线的索引。
有关完整详细信息,请参阅
diag_indices
。- 参数:
- arr数组,至少 2 维
另请参阅
注释
版本 1.4.0 中的新增功能。
示例
>>> 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]))