loop - python iterate numpy array

Numerar cómo iterar sobre las columnas de la matriz?

Question

Supongamos que tengo y mxn array. Quiero pasar cada columna de esta matriz a una función para realizar alguna operación en toda la columna. ¿Cómo iterar sobre las columnas de la matriz?

Por ejemplo, tengo una matriz 4 x 3 como

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

for column in array:
  some_function(column)

donde la columna sería "1,2,3,4" en la primera iteración, "99,14,12,43" en la segunda y "2,5,7,1" en la tercera.

Answer 1

Esto debería darte un comienzo

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


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

Answer 2

Para una matriz tridimensional puede intentar:

para c en array.transpose (1, 0, 2): do_stuff (c)

Consulte los documentos sobre cómo funciona array.transpose . Básicamente, estás especificando qué dimensión cambiar. En este caso, estamos cambiando la segunda dimensión (por ejemplo, columnas) a la primera dimensión.

Answer 3

Simplemente itere sobre la transposición de su matriz:

for column in array.T:
   some_function(column)

Answer 4

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