了两个二维数组

Question

numpy的迭代说我有两个矩阵：

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上的X两个相应值和Y，即形成：

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

有没有一种方法不使用双嵌套的for-loop？

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]) 

Answer 2

对于给定的numpy的阵列X和Y，你可能只是做 -

Zout = X**2 + Y**2 

如果你实际上是构建X和Y这样的，有一个直接的方式来获得Z与broadcasting从而避免np.meshgrid，像这样 -

Zout = np.arange(0, 2, 0.1)**2 + np.arange(3, 5, 0.1)[:,None]**2