python - numpy:将标量函数应用于不同矩阵的相应元素

Question

我有以下 4 个矩阵:

w = [[w1, w2, w3, w4], [w5, w6, w7, w8], ...]
x = [[x1, x2, x3, x4], [x5, x6, x7, x8], ...]
y = [[y1, y2, y3, x4], [y5, y6, y7, y8], ...]
z = [[z1, z2, z3, x4], [z5, z6, z7, z8], ...]

它们的形状为(n,m)。我想将标量函数 f 应用于每个矩阵的相应元素，使得所得矩阵 res 也是 n × m，并且 res 的第 i、j 个元素是 func([[w[i,j], x[i,j]], [[y[i,j], z[i,j]]) 的结果

显然我可以尝试循环遍历每个元素，但是是否有矢量化/numpy 实现？

Answer 1

创建一个 w、x、y、z 的广义函数，以向量化的方式逐元素应用它:http://docs.scipy.org/doc/numpy/reference/generated/numpy.vectorize.html