Numpy jak iterować po kolumnach tablicy?

Question

Załóżmy, że mam i tablicę m x n. Chcę przekazać każdą kolumnę tej tablicy do funkcji, aby wykonać operację na całej kolumnie. Jak mogę iterować po kolumnach tablicy?

Na przykład, posiada 4 x 3 tablicę jak

1 99 2 
2 14 5 
3 12 7 
4 43 1 

for column in array: 
    some_function(column) 

gdzie kolumna byłaby "1,2,3,4" w pierwszej iteracji "99,14,12,43" W drugi, a "2,5,7,1" w trzecim.

Answer 1

Wystarczy iteracyjne nad transponowane swojej tablicy:

for column in array.T: 
    some_function(column) 

Answer 2

To powinno dać początek

>>> for col in range(arr.shape[1]): 
    some_function(arr[:,col]) 


[1 2 3 4] 
[99 14 12 43] 
[2 5 7 1] 

Answer 3

for c in np.hsplit(array, array.shape[1]): 
    some_fun(c)