Numpy: 如何 map f: (shape (3) ndarray) --> (float) over an ndarray of shape (...,3) 以獲得 ndarray of shape (...)？

Question

我有一個 function 將形狀 (3) 的 ndarray 映射到浮點數，並且我有一個形狀 (...,3) 的 ndarray。 map function 在該陣列上獲得形狀陣列（...）的最佳方法是什么？

謝謝。

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)

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