Numpy如何迭代数组的列?

假设我有和mxn数组。 我想将这个数组的每一列传递给一个函数来对整个列执行一些操作。 如何遍历数组的列?

例如,我有一个4×3的数组

1 99 2 2 14 5 3 12 7 4 43 1 for column in array: some_function(column) 

在第一次迭代中列将是“1,2,3,4”,在第二次中是“99,14,12,43”,在第三次中是“2,5,7,1”。

只需遍历你的数组的转置:

 for column in array.T: some_function(column) 

这应该给你一个开始

 >>> for col in range(arr.shape[1]): some_function(arr[:,col]) [1 2 3 4] [99 14 12 43] [2 5 7 1] 
 for c in np.hsplit(array, array.shape[1]): some_fun(c)