Obtener columna de NumPy Array

Este tutorial presentará el método para obtener una columna específica de un array NumPy multidimensional en Python.

Obtenga una columna de un array NumPy multidimensional con el método de corte básico

El método de división básico funciona con el mismo principio que la división de listas en Python. Podemos usar el método de corte básico para obtener una columna específica de un array NumPy multidimensional. El método de corte básico crea una nueva vista de nuestra matriz existente en lugar de crear una nueva copia del array. Estos nuevos puntos de vista en el array existente y la memoria del array original no se pueden liberar hasta que el recolector de basura recopile todas las vistas que apuntan al array original. El siguiente ejemplo de código nos muestra cómo obtener una columna específica de un array NumPy multidimensional con el método de corte básico en Python.

import numpy as np

array = np.array([[1, 2], [3, 4], [5, 6], [7, 8], [9, 0]])
print(array[:, 1])

Producción :

[2 4 6 8 0]

En el código anterior, extrajimos la segunda columna del arraymPy multidimensional array con el índice de corte [:,1] en Python. La primera parte del índice es el índice de las filas. Hemos dejado la primera parte en blanco porque queremos seleccionar todas las filas. El operador : representa una operación de selección en el índice. Usamos [:,1] como índice de columna para obtener la segunda columna de cada fila.