Numpy：如何將f：（shape（3）ndarray） - >（float）映射到形狀（...，3）的狀態上以獲得形狀（...）的狀態？

Question

我有一個函數將一個形狀（3）的狀態映射到一個浮動狀態，並且我有一個形狀的狀態（...，3）。將該函數映射到該數組以獲取形狀（...）的數組的最佳方法是什麼？

謝謝。

Answer 1

你想要numpy.apply_along_axis。

def f(a): 
    return a[0] + a[1] + a[2] 
mm = numpy.random.randn(5, 3) 
numpy.apply_along_axis(f, 1, mm) 

輸出：array([-1.75875289, -0.34689792, 0.66092486, -0.21626001, -0.14125476])