python - 矢量化中的数组索引

Question

有没有办法在向量化的 numpy 方程中利用数组索引？

具体来说，我有这个循环代码，它将二维数组的每个值设置为到某个任意中心点的距离。

img=np.ndarray((size[0],size[1]))
for x in range(size[0]):
    for y in range(size[1]):
        img[x,y]=math.sqrt((x-center[0])**2+(y-center[1])**2)

我如何对其进行矢量化？

Answer 1

您可以使用广播轻松解决此问题:

import numpy as np

size = (64, 64)
center = (32, 32)

x = np.arange(size[0])
y = np.arange(size[1])

img = np.sqrt((x - center[0]) ** 2 + (y[:, None] - center[1]) ** 2)