應用一個函數逐元素到多個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

這取決於，真的。對於給定的功能;比如說'a + b'怎麼樣？大概你有更復雜的想法。

最一般的解決方案是np.vectorize;但它也是最慢的。根據你想要做什麼，儘管可能存在更聰明的解決方案。例如，看一下numexp。