Python:将函数应用于数组的每一列

Question

我需要对 numpy 数组的每一列应用一个函数。我不能对数组的每个元素都这样做，但它必须是每一列，因为每一列组合在一起代表一个信息。

import numpy as np
C = np.random.normal(0, 1, (500, 30))

这是最有效的方法吗(为了说明我正在使用 np.sum):

C2 = [ np.sum( C[ :, i ] )  for i in range( 0, 30) ]

数组 C 是 500x4000，我也在对每一列应用一个耗时的函数。

Answer 1

你可以试试np.apply_along_axis :

In [21]: A = np.array([[1,2,3],[4,5,6]])

In [22]: A
Out[22]: 
array([[1, 2, 3],
       [4, 5, 6]])

In [23]: np.apply_along_axis(np.sum, 0, A)
Out[23]: array([5, 7, 9])

In [24]: np.apply_along_axis(np.sum, 1, A)
Out[24]: array([ 6, 15])