🗑️ Как удалить столбец в массиве Python: простой гид

import numpy as np

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

column_index = 1  # индекс столбца, который нужно удалить

new_array = np.delete(array, column_index, axis=1)
print(new_array)
    

Как удалить столбец в массиве Python?

Удаление столбца в массиве Python может быть выполнено с помощью различных подходов, в зависимости от используемой библиотеки для работы с массивами. В этой статье мы рассмотрим два популярных способа удаления столбца из массива: с использованием библиотеки NumPy и с использованием библиотеки Pandas.

1. Удаление столбца с использованием NumPy

NumPy - это популярная библиотека для научных вычислений в Python. Она предоставляет функциональность для работы с многомерными массивами. Для удаления столбца из массива NumPy мы можем использовать функцию delete().

Вот простой пример удаления столбца из массива NumPy:

import numpy as np

# Создание двумерного массива
arr = np.array([[1, 2, 3],
                [4, 5, 6],
                [7, 8, 9]])

# Удаление столбца по индексу
new_arr = np.delete(arr, 1, axis=1)

print("Исходный массив:")
print(arr)

print("\nМассив после удаления столбца:")
print(new_arr)

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

2. Удаление столбца с использованием Pandas

Библиотека Pandas предоставляет более высокоуровневый и удобный интерфейс для работы с данными в Python. Для удаления столбца в Pandas мы можем использовать метод drop() объекта DataFrame.

Вот пример удаления столбца из массива Pandas:

import pandas as pd

# Создание DataFrame
df = pd.DataFrame({'A': [1, 2, 3],
                   'B': [4, 5, 6],
                   'C': [7, 8, 9]})

# Удаление столбца по имени
new_df = df.drop('B', axis=1)

print("Исходный DataFrame:")
print(df)

print("\nDataFrame после удаления столбца:")
print(new_df)

В этом примере мы создаем DataFrame df и используем метод drop(), указывая имя столбца, который нужно удалить (в данном случае столбец 'B') и ориентацию axis=1, чтобы указать, что мы хотим удалить столбец. Результатом будет новый DataFrame new_df без удаленного столбца.

Заключение

Удаление столбца в массиве Python может быть выполнено с помощью функции delete() из библиотеки NumPy или метода drop() из библиотеки Pandas. Оба подхода предоставляют возможность удалять столбцы и их значения из массива или DataFrame.

Не забудьте установить библиотеки NumPy и Pandas, если они еще не установлены, с помощью команды:

!pip install numpy pandas

Надеюсь, эта статья помогла вам понять, как удалить столбец в массиве Python.