Як видалити певні елементи з масиву numpy

Ви можете використовувати такі методи, щоб видалити певні елементи з масиву NumPy:

Спосіб 1: видалення елементів, що дорівнюють певному значенню

 #remove elements whose value is equal to 12
new_array = np. delete (original_array, np. where (original_array == 12))

Спосіб 2. Видалення зі списку елементів, що відповідають певному значенню

 #remove elements whose value is equal to 2, 5, or 12
new_array = np. setdiff1d (original_array, [2, 5, 12])

Спосіб 3: видалення елементів на основі позиції індексу

 #remove elements in index positions 0 and 6
new_array = np. delete (original_array, [0, 6])

Наступні приклади показують, як використовувати кожен метод на практиці.

Приклад 1: видалення елементів, що дорівнюють певному значенню

Наступний код показує, як видалити всі елементи з масиву NumPy, значення якого дорівнює 12:

 import numpy as np

#define original array of values
original_array = np. array ([1, 2, 2, 4, 5, 7, 9, 12, 12])

#remove elements whose value is equal to 12
new_array = np. delete (original_array, np. where (original_array == 12))

#view new array
print (new_array)

[1 2 2 4 5 7 9]

Зверніть увагу, що два елементи масиву, які дорівнювали 12, були видалені.

Приклад 2: Видалення зі списку елементів, що відповідають певному значенню

Наступний код показує, як видалити всі елементи з масиву NumPy, значення яких дорівнюють 2, 5 або 12:

 import numpy as np

#define original array of values
original_array = np. array ([1, 2, 2, 4, 5, 7, 9, 12, 12])

#remove elements whose value is equal to 2, 5, or 12
new_array = np. setdiff1d (original_array, [2, 5, 12])

#view new array
print (new_array)

[1 4 7 9]

Зауважте, що всі елементи зі значенням 2, 5 або 12 видалено.

Приклад 3: Видалення елементів на основі позиції індексу

Наступний код показує, як видалити елементи в позиціях індексу 0 і 6 з масиву NumPy:

 import numpy as np

#define original array of values
original_array = np. array ([1, 2, 2, 4, 5, 7, 9, 12, 12])

#remove elements in index positions 0 and 6
new_array = np. delete (original_array, [0, 6])

#view new array
print (new_array)

[2 2 4 5 7 12 12]

Зауважте, що елементи в позиції індексу 0 (зі значенням 1) і в позиції індексу 6 (зі значенням 9) обидва були видалені з масиву NumPy.

Додаткові ресурси

У наступних посібниках пояснюється, як виконувати інші типові операції в Python:

Як заповнити масив NumPy значеннями
Як замінити елементи в масиві NumPy
Як отримати певний рядок із масиву NumPy