Как удалить элемент из списка в numpy?

Пользователь

от lori_jast , в категории: Python , 2 года назад

Как удалить элемент из списка в numpy?

Facebook Vk Ok Twitter LinkedIn Telegram Whatsapp

2 ответа

Пользователь

от roxanne.hauck , 2 года назад

@lori_jast 

Вы можете удалить элемент из массива NumPy, используя метод numpy.delete(). Этот метод принимает массив и список индексов элементов, которые нужно удалить, и возвращает новый массив, в котором указанные элементы были удалены.


Например, если у вас есть массив a:

1
2
3
import numpy as np

a = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9])


Вы можете удалить элементы с индексами 2, 3 и 7 следующим образом:

1
2
3
b = np.delete(a, [2, 3, 7])

print(b)  # [1 2 4 5 6 8 9]


Обратите внимание, что этот метод возвращает новый массив, а исходный массив a остается неизменным. Если вы хотите удалить элементы из исходного массива, вы можете присвоить результат np.delete() обратно в a.

1
2
3
a = np.delete(a, [2, 3, 7])

print(a)  # [1 2 4 5 6 8 9]


Обратите внимание, что вы также можете удалить целые строки или столбцы из многомер

Пользователь

от steve , год назад

@lori_jast 

лярных массивов, передавая индексы строк или столбцов для удаления в качестве аргументов axis. Например, если у вас есть двумерный массив a:


1 2 3 4 5 6 7 8 9


a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])


Вы можете удалить вторую строку следующим образом:


1 2 3 4


b = np.delete(a, 1, axis=0)


print(b)

[[1 2 3]

[7 8 9]]

Аналогично, вы можете удалить второй столбец так:


1 2 3


c = np.delete(a, 1, axis=1)


print(c)

[[1 3]

[4 6]

[7 9]]

В этом примере, при удалении строки или столбца, указанный axis должен быть равен 0 или 1 соответственно.