🔥Как удалить колонку в pandas: простой и эффективный способ🔥

import pandas as pd

# Создание DataFrame
df = pd.DataFrame({'column_name': [1, 2, 3]})

# Удаление колонки "column_name"
df = df.drop('column_name', axis=1)
    

Как удалить колонку в pandas

Библиотека pandas предоставляет множество инструментов для работы с данными, включая возможность удаления колонок из DataFrame. Удаление колонок может быть полезным при очистке данных, обработке пропущенных значений или просто при необходимости отбросить ненужные признаки.

Вот несколько способов удалить колонку в pandas.

Метод drop()

Метод drop() позволяет удалить колонку из DataFrame.

import pandas as pd

# Создание DataFrame
data = {'Имя': ['Анна', 'Борис', 'Катя', 'Дмитрий'],
        'Возраст': [25, 32, 19, 45],
        'Город': ['Москва', 'Санкт-Петербург', 'Киев', 'Нью-Йорк']}

df = pd.DataFrame(data)

# Удаление колонки 'Город'
df = df.drop('Город', axis=1)

print(df)

Этот код создает DataFrame с тремя колонками: 'Имя', 'Возраст' и 'Город'. Затем метод drop() используется для удаления колонки 'Город' с помощью параметра axis=1. Значение параметра axis=1 указывает, что мы хотим удалить колонку, а не строку. Наконец, измененный DataFrame выводится на экран.

Метод drop() также позволяет удалить несколько колонок одновременно, передав их имена в виде списка:

# Удаление колонок 'Возраст' и 'Город'
df = df.drop(['Возраст', 'Город'], axis=1)

print(df)

Использование ключевого слова del

В pandas также можно использовать ключевое слово del для удаления колонки из DataFrame. Однако, при использовании del необходимо обратиться к DataFrame по имени колонки. Вот пример:

# Удаление колонки 'Город' с использованием del
del df['Город']

print(df)

В этом примере колонка 'Город' удаляется из DataFrame с помощью ключевого слова del.

Использование метода pop()

Метод pop() позволяет удалить колонку из DataFrame и вернуть ее в виде отдельного Series.

# Удаление колонки 'Возраст' с использованием метода pop()
age = df.pop('Возраст')

print(df)
print(age)

В этом примере колонка 'Возраст' удаляется из DataFrame с использованием метода pop(). Затем удаленная колонка сохраняется в переменную age и выводится на экран как отдельный Series.

Использование метода drop()

Метод drop() также может быть использован для удаления строк из DataFrame. Для удаления строк используется параметр axis=0. Вот пример:

# Удаление строк с индексами 0 и 1
df = df.drop([0, 1], axis=0)

print(df)

Этот код удаляет первые две строки из DataFrame с помощью метода drop() и параметра axis=0.

Заключение

В этой статье мы рассмотрели несколько способов удаления колонок в pandas. Методы drop(), del и pop() предоставляют различные возможности для удаления колонок из DataFrame в зависимости от ваших потребностей.

Имейте в виду, что эти методы изменяют исходный DataFrame. Если вы хотите сохранить исходный DataFrame в неизменном виде, рекомендуется создать его копию перед удалением колонок.