python - numpy 遍历两个二维数组

Question

假设我有两个矩阵:

X, Y = np.meshgrid(np.arange(0, 2, 0.1), np.arange(3, 5, 0.1))

还有一个函数，比如:

def func(x) :
    return x[0]**2 + x[1]**2

如何填充矩阵 Z(大小为 np.shape(X))，其中每个条目都是通过对 的两个对应值调用 func 形成的code>X 和 Y，即:

Z[i, j] = func([X[i, j], Y[i, j]])

有没有不使用双嵌套 for 循环的方法？

Answer 1

这也是函数评估的向量化形式:

import numpy as np
X, Y = np.meshgrid(np.arange(0, 2, 0.1), np.arange(3, 5, 0.1))
def func(x) :
    return x[0]**2 + x[1]**2

Z = func([X,Y])