python - 在网格的每个点创建 numpy 数组

Question

我正在寻找一种矢量化方法来在 2D 网格的每个点创建 2D 数组，并将结果存储在 4D 数组中（或者至少比嵌套循环更好）。

更具体地说，我有一个函数可以在给定两个数字的情况下创建一个矩阵（二维数组）H(x,y)。我想使用这个函数在(x,y)网格的每个点创建一个矩阵，并将结果存储在一个 4D 数组中。最好，如果我们有类似的东西

x = np.linspace(-1,1,100)
y = np.linspace(-1,1,100)

xx, yy = np.meshgrid(x,y)

result = magic_function(xx, yy, H)

那么 的形状result将是(100,100,4,4)（如果H正在制作 4x4 矩阵）。

标签： pythonarraysnumpymultidimensional-array