Loop em uma matriz do NumPy
Ao lidar com uma matriz 1D do NumPy, um loop por todos os elementos pode ser algo simples, como:
for x in my_array :
...
Ao lidar com uma matriz 2D do NumPy, a situação é mais complicada. Uma matriz 2D é formada por várias matrizes 1D. Para iterar de maneira explícita por todos os elementos separados de uma matriz multidimensional, é necessário usar esta sintaxe:
for x in np.nditer(my_array) :
...
Duas matrizes do NumPy que você talvez reconheça do curso de introdução estão disponíveis em sua sessão do Python: np_height
, uma matriz do NumPy que contém a altura dos jogadores da Major League Baseball, e np_baseball
, uma matriz 2D do NumPy que contém a altura (primeira coluna) e o peso (segunda coluna) desses jogadores.
Este exercício faz parte do curso
Python intermediário
Instruções de exercício
- Importe o pacote
numpy
com o nome localnp
. - Escreva um loop
for
que itere por todos os elementos denp_height
e imprima"x inches"
para cada elemento, em que x é o valor na matriz. - Escreva um loop
for
que acesse cada elemento da matriznp_baseball
e o imprima.
Exercício interativo prático
Experimente este exercício preenchendo este código de exemplo.
# Import numpy as np
# For loop over np_height
# For loop over np_baseball