Session Ready
Exercise

Selección de array 2D con NumPy

Si tu array numpy tiene una estructura regular, por ejemplo, cada fila y columna tiene un valor fijo de valores, formas complicadas de selección se vuelven muy fáciles. Mira el código de abajo, donde los elementos "a" y "c" son extraídos de una lista de listas.

# Una lista de listas común
x = [["a", "b"], ["c", "d"]]
[x[0][0], x[1][0]]

# numpy
import numpy as np
np_x = np.array(x)
np_x[:,0]

Para una lista común de Python, esto es un dolor de cabeza. Para arrays 2D numpy, ¡El tema se vuelve más intuitivo!. Los índices antes de la coma se refieren a las filas mientras que los que le siguen se refieren a las columnas. El : es utilizado para seleccionar; en este ejemplo, le dice a Python que selecciona todas las filas.

El código que convierte la lista precargada baseball en un array 2D numpy ya se encuentra disponible en el guión. La primera columna contiene la altura de los jugadores en pulgadas y la segunda columna contiene el peso de los jugadores en libras. Agrega algunas líneas de código para hacer las respectivas correciones. Recuerda que en Python, ¡El primer elemento está en el índice 0!

Instructions
100 XP
  • Imprime la fila N°50 de np_baseball.
  • Crea una nueva variable, np_weight, la cual contiene toda la segunda columna de np_baseball.
  • Selecciona la altura (primera columna) del jugador N°124 en np_baseball e imprimelo.