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)