Python Numpy：如何映射f：（shape（3）ndarray）-->；（浮动）在形状的数组（…，3）上以获得形状的数组（…）？

我有一个函数，它把一个数组的形状（3）映射到一个浮点，我有一个数组的形状（…，3）。将该函数映射到该数组以获得形状数组（…）的最佳方法是什么

---

谢谢。

您需要

numpy。沿\u轴应用\u

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

---

输出：

数组（[-1.75875289，-0.34689792，0.66092486，-0.21626001，-0.14125476]）

您需要

numpy。沿_轴应用

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

输出：

数组（[-1.75875289，-0.34689792,0.66092486，-0.21626001，-0.14125476]）