Indexação de array NumPy

Este tutorial irá apresentar os métodos para especificar o índice do array NumPy.

Indexação de array NumPy

Array indexing é usado para acessar elementos especificando seus índices dentro do array. Se tivermos um array preenchido com zeros e quisermos colocar um valor particular em um índice específico dentro do array, podemos usar o método de indexação de array. A indexação de array funciona de maneira muito diferente para matrizes 1D e 2D em Python. Se quisermos acessar os dois primeiros elementos do array 2D como fizemos com o array 1D, temos que usar o índice Array[(0,1),(0,1)].

import numpy as np

matrix = np.zeros((3, 3))

values = np.array([1, 2, 3])

matrix[(0, 1, 2), (0, 1, 2)] = values
print(matrix)

Resultado:

[[1. 0. 0.]
 [0. 2. 0.]
 [0. 0. 3.]]

Substituímos os zeros em índices específicos em nossa matriz matriz pelos valores dentro do array values usando a indexação de array NumPy. Primeiro criamos um array matriz e a preenchemos com zeros. Em seguida, criamos a matriz values contendo valores que queremos inserir em nossa matriz. Em seguida, acessamos os valores dentro do array com matriz [(0,1,2), (0,1,2)] = valores. Substitui os valores nos índices 0,0, 1,1 e 2,2 da matriz.