Subsetting av 2D NumPy-arrayer
Om din 2D-array i numpy har en regelbunden struktur – det vill säga att varje rad och kolumn har ett fast antal värden – blir avancerade sätt att ta ut delmängder mycket enkla. Titta på koden nedan där elementen "a" och "c" hämtas ut från en lista av listor.
# numpy
import numpy as np
np_x = np.array(x)
np_x[:, 0]
Indexen före kommatecknet avser raderna, medan de efter kommatecknet avser kolumnerna. Tecknet : används för slicing; i det här exemplet anger det att Python ska inkludera alla rader.
Den här övningen är en del av kursen
Introduktion till Python
Övningsinstruktioner
- Skriv ut den 50:e raden i
np_baseball. - Skapa en ny variabel,
np_weight_lb, som innehåller hela den andra kolumnen inp_baseball. - Välj ut längden (första kolumnen) för den 124:e basebollspelaren i
np_baseballoch skriv ut den.
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
import numpy as np
np_baseball = np.array(baseball)
# Print out the 50th row of np_baseball
# Select the entire second column of np_baseball: np_weight_lb
# Print out height of 124th player