在Python中对2d numpy数组中的每个向量应用操作

Question

我需要它的二维阵列，但也有兴趣在更高的维度。 如果我有两个numpy2darray，比如A和B，还有一个函数f，它取2个向量并输出一个标量。性能很重要，而且numpy数组操作非常快，因此如何获得矩阵：

(f(a1,b1),f(a1, b2)...)
(f(a2,b1), f(a2,b2)...)
(...                  )

a（i）和b（i）是一维数组/向量

通过应用类似于f（A，B）的东西，而不使用任何循环

有了一些方法，我可以使用axis=1，但是我该如何定义自己的函数呢？或者还有其他方法吗

Answer 1

最简单、最有效的解决方案是使用for循环。矢量化在这里没有帮助，因为它将标量传递给函数

假设数组有两列：

for a,b in zip(A,B):
    print(f(a[0],b[0]))
    print(f(a[1],b[1]))

将打印要返回的f值