Como fatiar um array numpy 2d (com exemplos)

Você pode usar os seguintes métodos para fatiar uma matriz NumPy 2D:

Método 1: selecione linhas específicas no array NumPy 2D

 #select rows in index positions 2 through 5
arr[ 2 : 5 ,:]

Método 2: selecionar colunas específicas na tabela NumPy 2D

 #select columns in index positions 1 through 3
arr[:, 1 : 3 ]

Método 3: selecione linhas e colunas específicas na tabela NumPy 2D

 #select rows in range 2:5 and columns in range 1:3
arr[ 2 : 5,1 :3 ]

Os exemplos a seguir mostram como usar cada método na prática com o seguinte array NumPy 2D:

 import numpy as np

#create NumPy array
arr = np. arange ( 24 ). reshape ( 6,4 )

#view NumPy array
print (arr)

[[ 0 1 2 3]
 [4 5 6 7]
 [8 9 10 11]
 [12 13 14 15]
 [16 17 18 19]
 [20 21 22 23]]

Exemplo 1: Selecione linhas específicas de uma matriz NumPy 2D

Podemos usar a seguinte sintaxe para selecionar linhas nas posições de índice 2 a 5:

 #select rows in index positions 2 through 5
arr[ 2 : 5 ,:]

array([[ 8, 9, 10, 11],
       [12, 13, 14, 15],
       [16, 17, 18, 19]])

Observe que a sintaxe 2:5 diz ao NumPy para selecionar as linhas 2 a 5, mas não inclui 5.

Portanto, esta sintaxe seleciona todos os valores das linhas com posições de índice 2, 3 e 4.

Exemplo 2: Selecione colunas específicas da matriz NumPy 2D

Podemos usar a seguinte sintaxe para selecionar colunas nas posições de índice 1 a 3:

 #select columns in index positions 1 through 3
arr[, 1 : 3 ]

array([[ 1, 2],
       [5, 6],
       [9, 10],
       [13, 14],
       [17, 18],
       [21, 22]]))

Observe que a sintaxe 1:3 diz ao NumPy para selecionar as colunas 1 a 3, mas não inclui 3.

Portanto, esta sintaxe seleciona todos os valores das colunas com posições de índice 1 e 2.

Exemplo 3: Selecione linhas e colunas específicas do array NumPy 2D

Podemos usar a seguinte sintaxe para selecionar linhas nas posições de índice 2 a 5 e colunas nas posições de índice 1 a 3:

 #select rows in 2:5 and columns in 1:3
arr[ 2 : 5,1 :3 ]

array([[ 9, 10],
       [13, 14],
       [17, 18]])

Esta sintaxe retorna todos os valores na matriz NumPy 2D entre as posições de índice de linha 2 a 5 e as posições de índice de coluna 1 a 3.

Recursos adicionais

Os tutoriais a seguir explicam como realizar outras operações comuns no NumPy:

Como encontrar o índice de valor na matriz NumPy
Como obter uma coluna específica de um array NumPy
Como adicionar uma coluna a um array NumPy