python - 基于另一个数组更改 numpy 数组值的矢量化方法

Question

是否有一种向量化(或更好)的方式将值设置为特定值numpy 数组的数据点基于除此之外的另一种方式？

import numpy as np

data = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
pos = np.array([[1, 2], [2, 0]])

for p in pos:
    i,j = p
    data[i,j] = 20

print(data)

Answer 1

使用更高版本的 Python，您可以通过解包另一个可迭代对象来在推导式中创建列表。然后我们传递该列表以进行切片分配。

我们访问(切片)的方式是通过 Integer Array Indexing 完成的

data[[*pos]] = 20
data

array([[ 1,  2,  3],
       [ 4,  5, 20],
       [20,  8,  9]])

---

对于其他版本的 Python 尝试:

data[pos.tolist()] = 20
data