Como substituir elementos em um array numpy (3 exemplos)

Você pode usar os seguintes métodos para substituir elementos em uma matriz NumPy:

Método 1: Substitua elementos iguais a um determinado valor

 #replace all elements equal to 8 with a new value of 20
my_array[my_array == 8 ] = 20

Método 2: Substitua itens com base em uma condição

 #replace all elements greater than 8 with a new value of 20
my_array[my_array > 8 ] = 20

Método 3: substituir elementos com base em múltiplas condições

 #replace all elements greater than 8 or less than 6 with a new value of 20
my_array[(my_array > 8 ) | (my_array < 6 )] = 20

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

 import numpy as np

#create array
my_array = np. array ([4, 5, 5, 7, 8, 8, 9, 12])

#view array
print (my_array)

[4 5 5 7 8 8 9 12]

Método 1: Substitua elementos iguais a um determinado valor

O código a seguir mostra como substituir todos os elementos do array NumPy iguais a 8 por um novo valor de 20 :

 #replace all elements equal to 8 with 20
my_array[my_array == 8 ] = 20

#view updated array
print (my_array)

[4 5 5 7 20 20 9 12]

Método 2: Substitua itens com base em uma condição

O código a seguir mostra como substituir todos os elementos da matriz NumPy maiores que 8 por um novo valor de 20 :

 #replace all elements greater than 8 with 20
my_array[my_array > 8 ] = 20

#view updated array
print (my_array)

[4 5 5 7 8 8 20 20]

Método 3: substituir elementos com base em múltiplas condições

O código a seguir mostra como substituir todos os elementos da matriz NumPy maiores que 8 ou menores que 6 por um novo valor de 20 :

 #replace all elements greater than 8 or less than 6 with a new value of 20
my_array[(my_array > 8 ) | (my_array < 6 )] = 20

#view updated array
print (my_array)

[20 20 20 7 8 8 20 20]

Recursos adicionais

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

Como calcular o modo do array NumPy
Como encontrar o índice de valor na matriz NumPy
Como mapear uma função para um array NumPy