Arrays com Python + Numpy

Fernando Feltrin · Capítulo 20 de 36

Páginas do PDF

Arrays com Python + Numpy

Consultando elementos dentro de um intervalo

data2 = np.random.randint ( 5 , size = ( 3 , 4 ))

print ( data2 )

print ( data2 [ 0 , 2 ])

 

Note que é criada a array data2, de elementos aleatórios entre 0 e 5, tendo 3 linhas e 4 colunas em sua forma. Parametrizando print com data2[0, 2] estamos pedindo que seja exibido o elemento situado na linha 0 (primeira linha) e coluna 2 (terceira coluna pois a primeira coluna é 0).

 

Podemos visualizar via console a array em si, e o respectivo elemento situado na primeira linha, terceira coluna. Neste caso, o número 4.

 

Consultando elementos dentro de um intervalo

data

print ( data )

 

Reutilizando nossa array data criada anteriormente, uma simples array unidimensional de 15 elementos ordenados, podemos avançar com o entendimento de outras formas de consultar dados/valores, dessa vez, definindo intervalos específicos.

print ( data )

print ( data [: 5 ])

print ( data [ 3 :])

print ( data [ 4 : 8 ])

print ( data [:: 2 ])

print ( data [ 3 ::])

 

Como mencionado anteriormente, a forma como podemos consultar elementos via índice é muito parecida (senão igual) a forma como realizávamos as devidas consultas em elementos de uma lista. Sendo assim, podemos usar das mesmas notações aqui.

Na primeira linha simplesmente por meio da função print( ) estamos exibindo em tela os dados contidos em data.

Na sequência, passando data[:5] (mesmo que data[0:5]) como parâmetro para nossa função print( ) estaremos exibindo os 5 primeiros elementos dessa array. Em seguida via data[3:] estaremos exibindo do terceiro elemento em diante todos os demais. Na sequência, via data[4:8] estaremos exibindo do quarto ao oitavo elemento.

Da mesma forma, porém com uma pequena diferença de notação, podemos por meio de data[::2] exibir de dois em dois elementos, todos os elementos. Por fim, via data[3::] estamos pulando os 3 primeiros elementos e exibindo todos os demais.

 

Como esperado, na primeira linha temos todos os dados da array, na segunda linha apenas os 5 primeiros, na terceira linha do terceiro elemento em diante, na quarta linha os elementos entre 4 e 8, na quinta linha os elementos pares e na sexta linha todos os dados exceto os três primeiros.