python - 向量化数组的行

Question

我有一个数组 X，我想将函数 f 应用于 X 的所有行:

# silly example
X = numpy.array([[1, 2, 3, 4, 5],
                 [6, 7, 8, 9, 0]], 'i')

def f(row): return sum(row)

y = numpy.vectorize(f, 'i')(rows(X))

现在，y 应该是 array([15,30], 'i')。哪种方法或切片魔术将以最有效的方式实现 rows？

Answer 1

NumPy 实现了“特定轴上的 Action ”的概念。一般函数是numpy.apply_along_axis() :

>>> numpy.apply_along_axis(sum, 1, X)
array([15, 30])

(其中 sum 当然可以用任何东西代替)。