小编gud*_*ife的帖子

如何获得numpy数组的非对角元素的索引?

如何获得numpy数组的非对角元素的索引?

a = np.array([[7412, 33, 2],
              [2, 7304, 83],
              [3, 101, 7237]])
Run Code Online (Sandbox Code Playgroud)

我尝试如下:

diag_indices = np.diag_indices_from(a)
print diag_indices
(array([0, 1, 2], dtype=int64), array([0, 1, 2], dtype=int64))
Run Code Online (Sandbox Code Playgroud)

在那之后,不知道......预期的结果应该是:

result = [[False, True, True],
        [True, False, True],
         [True, True, False]]
Run Code Online (Sandbox Code Playgroud)

python numpy

7
推荐指数
1
解决办法
4509
查看次数

标签 统计

numpy ×1

python ×1