Numpy comment itérer sur les colonnes du tableau ?

Question

Supposons que j'aie et m x n tableau. Je veux passer chaque colonne de ce tableau à une fonction pour effectuer une opération sur la colonne entière. Comment puis-je itérer sur les colonnes du tableau ?

Par exemple, j'ai un tableau 4 x 3 comme

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

for column in array:
  some_function(column)

où colonne serait "1,2,3,4" dans la première itération, "99,14,12,43" dans la deuxième, et "2,5,7,1" dans la troisième.

Answer 1

Il suffit d'itérer sur la transposition de votre tableau :

for column in array.T:
   some_function(column)

Answer 2

Cela devrait vous donner un début

>>> 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

Vous pouvez également utiliser unzip pour itérer à travers les colonnes

for col in zip(*array):
   some_function(col)

Answer 4

Pour un tableau tridimensionnel, vous pouvez essayer :

for c in array.transpose(1, 0, 2):
    do_stuff(c)

Consultez les documents sur le fonctionnement array.transpose. Fondamentalement, vous spécifiez quelle dimension à décaler. Dans ce cas, nous déplaçons la deuxième dimension (par exemple les colonnes) vers la première dimension.

Answer 5

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