如何用另一个带索引的数组替换一个 numpy 数组中的值

Question

我有两个 arrays 和一个带索引的列表

阵列源（阵列01）：

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

具有新值的数组（数组 02）：

y = np.array([11, 22, 33])
array([11, 22, 33])

和一个索引列表

a = [1,2,1]
[1, 2, 1]

我需要根据索引替换数组 01 中数组 02 的值

预期回报：

array([[1, 11, 3],
       [4, 5, 22],
       [7, 33, 9]])

如何在没有循环的情况下做到这一点？

提前致谢。

Answer 1

尝试高级索引：

x[np.arange(len(x)), a] = y

我还发现本教程对我来说比官方文档更容易。