Python numpy 使用另一个数组将值分配给二维数组，其中有序对作为二维数组的索引

Question

这很难用语言描述，但在实践中很容易看到。我有一个二维数组：

im = np.array([[0,0,0,0],[0,0,0,0],[0,0,0,0],[0,0,0,0]])

我将其解释为 4x4 灰度图像 - 因此数组中的值只是强度。所以，首先，im 是：

[[0,0,0,0],
 [0,0,0,0],
 [0,0,0,0],
 [0,0,0,0]]

我希望能够根据 x 值数组和 y 值数组一次更改“图像”中的许多值。我将它们组装成第二个数组中的有序对，如下所示：

x = [0,1]
y = [2,3]
coords = np.array([x,y]).T

现在coords 看起来像这样：

array([[0, 2],
       [1, 3]])

最后，我想按坐标索引 im。我想也许是这样的：

im[coords] = 9

...但这不起作用。我希望 im 的最终结果是：

[[0,0,9,0],
 [0,0,0,9],
 [0,0,0,0],
 [0,0,0,0]]

有谁知道这样做的快速而优雅的方式？

谢谢！

Answer 1

你可以简单地做im[x,y] = 9。

例如：

im = np.array([[0,0,0,0],[0,0,0,0],[0,0,0,0],[0,0,0,0]])
x = [0,1]
y = [2,3]
im[x,y] = 9
print(im)
# Result:
# array([[0, 0, 9, 0],
#        [0, 0, 0, 9],
#        [0, 0, 0, 0],
#        [0, 0, 0, 0]])

Answer 2

一般来说，如果你有一个 numpy 数组

import numpy as np
arr = np.array([[0,0,0,0],[0,0,0,0],[0,0,0,0],[0,0,0,0]])
x_coords = [0, 1]
y_coords = [2, 3]
values = [8, 9]

然后

arr[x, y] = values

会导致

  array([
        [0, 0, 8, 0],
        [0, 0, 0, 9],
        [0, 0, 0, 0],
        [0, 0, 0, 0]
        ])