将函数元素应用于多个numpy数组

Question

假设我有两个相同尺寸的numpy数组，例如：

a = np.ones((4,))
b = np.linspace(0,4,4)

和一个应该对这些数组的元素进行操作的函数：

def my_func (x,y):
   # do something, e.g.
   z = x+y
   return z

如何以元素方式将此函数应用于a和b的元素并获得结果？

Answer 1

这取决于，真的。对于给定的功能;例如＆＃39; a + b＆＃39;怎么样？据推测，你有一些更复杂的想法。

最通用的解决方案是np.vectorize;但它也是最慢的。根据您的想法，可能会有更聪明的解决方案。例如，看看numexp。