如何更改多次索引的数组的值?

时间:2019-05-25 15:22:18

标签: arrays python-2.7 numpy

让我们像下面这样A:

A = np.array([[0, 0, 0, 0, 0],
              [0, 1, 0, 0, 0],
              [1, 0, 0, 0, 0],
              [1, 0, 1, 0, 0],
              [0, 0, 0, 0, 0]])

现在,我必须将第一列中所有带有1的行:

A[A[:, 0] > 0]
return: array([[1, 0, 0, 0, 0],
               [1, 0, 1, 0, 0]])

现在,我要修改第二行的第一列吗?我该怎么做 ?因为这不起作用:

A[A[:, 0] > 0][1, 1] = 1
A
array([[0, 0, 0, 0, 0],
       [0, 1, 0, 0, 0],
       [1, 3, 0, 0, 0],
       [1, 0, 1, 0, 0],
       [0, 0, 0, 0, 0]])

我的代码比这更复杂,因此条件必须分两次进行,例如:(1)> 0和(2)[1,1]。

0 个答案:

没有答案