¿Cómo establecer ejes para filas y columnas en NumPy?

En este artículo, vamos a ver cómo configurar el eje para filas y columnas en NumPy.

Funciones utilizadas

• np.array(objeto): para crear una array NumPy, el objeto es el parámetro que contiene la array
• np.reshape(filas, columnas): para remodelar la array en el número especificado de filas y columnas. Aquí, en los siguientes ejemplos, hemos dado -1 en lugar de filas para que numpy descubra si hay 3 columnas en cada fila.
• np.sum(axis): para calcular la suma o suma de los elementos. Aquí hemos mencionado el eje para realizar la operación en forma de array, fila o columna según el requisito.

Ejemplo 1: establecer el eje para el cálculo en forma de array

En este ejemplo, reformaremos la array NumPy en filas con 3 columnas cada una, es decir, nparray.reshape(-1, 3) para que sea bidimensional. Luego, realizaremos la operación de suma de los elementos de la array en orden normal, comenzando desde el primer hasta el último elemento de la array NumPy. Configuramos específicamente el eje = Ninguno para activar la operación normal de array.

Código:

import numpy as np
nparray = np.array([[1, 2, 3], [11, 22, 33],
                    [4, 5, 6], [8, 9, 10], 
                    [20, 30, 40]])
  
nparray = nparray.reshape(-1, 3)
print(nparray)
  
# calculating sum along 
# axix=None i.e array-wise
output = nparray.sum(axis=None)
print("\n\nSum array-wise: ", output)

Producción :

[[ 1  2  3]
 [11 22 33]
 [ 4  5  6]
 [ 8  9 10]
 [20 30 40]]


Sum array-wise:  204

Ejemplo 2: establecer el eje para el cálculo por columnas

En este ejemplo, remodelaremos la array numpy en filas con 3 columnas cada una. Luego realice la operación de suma de los elementos de la array usando la función sum() en forma de columna. Establecemos específicamente el eje = 0 para activar la operación normal de array.

Código:

import numpy as np
  
  
nparray = np.array([[1, 2, 3], [11, 22, 33],
                    [4, 5, 6], [8, 9, 10],
                    [20, 30, 40]])
nparray = nparray.reshape(-1, 3)
print(nparray)
  
# calculating sum along axix=0 
# i.e column-wise
output = nparray.sum(axis = 0)
print("\n\nSum column-wise: ", output)

Producción :

[[ 1  2  3]
 [11 22 33]
 [ 4  5  6]
 [ 8  9 10]
 [20 30 40]]


Sum column-wise:  [44 68 92]

Ejemplo 3: establecer el eje para el cálculo por filas

Estableceremos específicamente el eje = 1 para activar el cálculo normal por filas.

Código:

import numpy as np
nparray = np.array([[1, 2, 3], [11, 22, 33],
                    [4, 5, 6], [8, 9, 10], 
                    [20, 30, 40]])
  
nparray = nparray.reshape(-1, 3)
print(nparray)
  
# calculating sum along axix=1 
# i.e row0wise
output = nparray.sum(axis = 1)
print("\n\nSum row-wise: ", output)

Producción :

[[ 1  2  3]
 [11 22 33]
 [ 4  5  6]
 [ 8  9 10]
 [20 30 40]]


Sum row-wise:  [ 6 66 15 27 90]