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

Чтобы удалить элементы из массива numpy, можно использовать функцию numpy.delete(). Она принимает массив и позиции элементов, которые нужно удалить, и возвращает новый массив без удаленных элементов.

import numpy as np

# Создаем исходный массив
arr = np.array([1, 2, 3, 4, 5])

# Удаляем элемент с индексом 2
new_arr = np.delete(arr, 2)

print(new_arr) # [1 2 4 5]

Нumpy: как удалить элементы

Привет! В этой статье мы рассмотрим, как удалить элементы в библиотеке NumPy. NumPy предоставляет мощные инструменты для манипуляции с массивами данных, включая удаление элементов.

Удаление элементов из одномерного массива

Для удаления элементов из одномерного массива в NumPy вы можете использовать функцию delete(). Эта функция принимает три аргумента: сам массив, индексы элементов, которые нужно удалить, и ось, по которой нужно произвести удаление. Если ось не указана, элементы будут удалены плоско.

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
new_arr = np.delete(arr, [1, 3])  # Удаляем элементы с индексами 1 и 3

print(new_arr)  # [1 3 5]

В приведенном выше примере мы создаем одномерный массив arr с элементами [1, 2, 3, 4, 5]. Затем мы используем функцию delete(), чтобы удалить элементы с индексами 1 и 3. Результатом будет массив new_arr с оставшимися элементами [1, 3, 5].

Удаление элементов из многомерного массива

Если у вас есть многомерный массив, то NumPy позволяет вам удалить элементы по определенной оси.

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
new_arr = np.delete(arr, 1, axis=0)  # Удаляем вторую строку (с индексом 1)

print(new_arr)
# [[1 2 3]
#  [7 8 9]]

В приведенном выше примере у нас есть двумерный массив arr с элементами:

[[1 2 3]
 [4 5 6]
 [7 8 9]]

Мы использовали функцию delete() с аргументами arr, 1, axis=0. Это означает, что мы хотим удалить вторую строку (с индексом 1) по оси 0 (строки). Результатом будет массив new_arr:

[[1 2 3]
 [7 8 9]]

Вы также можете использовать функцию delete() для удаления элементов из других осей многомерного массива, изменяя значение аргумента axis.

Заключение

Теперь вы знаете, как удалить элементы из массива с помощью библиотеки NumPy. Используя функцию delete(), вы можете удалять элементы как из одномерных, так и из многомерных массивов. Удачи в ваших экспериментах с NumPy!