python - 输入数组相等时索引矩阵元素

Question

我有一个矩阵，我希望能够在不使用循环的情况下使用两个数组对某些元素进行索引时更改它们的值

例如

import numpy as np

A = np.array([[0, 0, 0, 0],
              [0, 0, 0, 0],
              [0, 0, 0, 0]])
b = np.array([0, 1, 2])
c = np.array([2, 3, 1, 1])

如果 b 中的索引 i 等于 c 中的索引 j 则访问 A[i,j]

我试过了

A[b, c] = 1

和

A[b == c] = 1

但我没有得到正确的结果我相信结果应该是

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

Answer 1

您可以使用 numpy.meshgrid 将一维索引数组扩展为二维索引数组:

import numpy as np

A = np.array([[0, 0, 0, 0],
              [0, 0, 0, 0],
              [0, 0, 0, 0]])
b = np.array([0, 1, 2])
c = np.array([2, 3, 1, 1])

C,B = np.meshgrid(c,b)
A[B==C] = 1

print(A)

给予

[[0 0 0 0]
 [0 0 1 1]
 [1 0 0 0]]