Indexação e fatiamento de arrays

Introdução ao NumPy

Izzy Weber

Core Curriculum Manager, DataCamp

Indexando arrays 1D

array = np.array([2, 4, 6, 8, 10])
array[3]

Indexando elementos em 2D

Um jogo de sudoku com um único número selecionado na terceira linha da quinta coluna

sudoku_game[2, 4]

Indexando linhas em 2D

Um jogo de sudoku com a primeira linha selecionada

sudoku_game[0]

array([0, 0, 4, 3, 0, 0, 2, 0, 9])

Indexando colunas em 2D

Um jogo de sudoku com a quarta coluna selecionada

sudoku_game[:, 3]

array([3, 0, 0, 0, 0, 0, 0, 5, 9])

Fatiando arrays 1D

array = np.array([2, 4, 6, 8, 10])
array[2:4]

array([6,  8])

Fatiando arrays 2D

Um jogo de sudoku com os nove quadrados do meio selecionados - colunas quatro a seis e linhas quatro a seis

sudoku_game[3:6, 3:6]

array([[0, 0, 2],
       [0, 0, 7],
       [0, 8, 3]])

Fatiando com passos

Um jogo de sudoku com apenas os cantos dos nove quadrados do meio selecionados

sudoku_game[3:6:2, 3:6:2]

array([[0, 2],
       [0, 3]])

Ordenando arrays

Antes de ordenar: Um jogo de sudoku não ordenado

np.sort(sudoku_game)

array([[0, 0, 0, 0, 0, 2, 3, 4, 9],
       [0, 0, 0, 0, 0, 0, 1, 5, 9],
       [0, 0, 0, 0, 0, 3, 4, 6, 7],
       [0, 0, 0, 0, 0, 2, 6, 7, 8],
       [0, 0, 0, 0, 0, 1, 4, 7, 9],
       [0, 0, 0, 0, 0, 0, 3, 5, 8],
       [0, 0, 0, 0, 0, 0, 1, 5, 6],
       [0, 0, 0, 0, 3, 5, 6, 8, 9],
       [0, 0, 0, 0, 1, 2, 3, 4, 9]])

Ordem dos eixos

gráfico mostrando que o eixo zero é a direção ao longo das linhas e o eixo um é a direção ao longo das colunas

Gráfico mostrando que o número um parece uma coluna

Ordenando por eixo

np.sort(sudoku_game)

array([[0, 0, 0, 0, 0, 2, 3, 4, 9],
       [0, 0, 0, 0, 0, 0, 1, 5, 9],
       [0, 0, 0, 0, 0, 3, 4, 6, 7],
       [0, 0, 0, 0, 0, 2, 6, 7, 8],
       [0, 0, 0, 0, 0, 1, 4, 7, 9],
       [0, 0, 0, 0, 0, 0, 3, 5, 8],
       [0, 0, 0, 0, 0, 0, 1, 5, 6],
       [0, 0, 0, 0, 3, 5, 6, 8, 9],
       [0, 0, 0, 0, 1, 2, 3, 4, 9]])

np.sort(sudoku_game, axis=0)

array([[0, 0, 0, 0, 0, 0, 0, 0, 0],
       [0, 0, 0, 0, 0, 0, 0, 0, 0],
       [0, 0, 0, 0, 0, 0, 0, 0, 0],
       [0, 0, 0, 0, 0, 0, 0, 0, 0],
       [0, 0, 2, 0, 0, 2, 1, 0, 1],
       [0, 4, 3, 0, 0, 3, 2, 0, 3],
       [0, 5, 4, 3, 1, 7, 3, 4, 5],
       [1, 7, 5, 5, 6, 8, 4, 8, 7],
       [6, 9, 6, 9, 8, 9, 6, 9, 9]])

Vamos praticar!

Introdução ao NumPy