Question

我有一个将形状（3）的ndarray映射到浮点数的函数，我有一个形状的ndarray（...，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])