Как удалить колонку в pandas dataframe: простой способ с помощью метода drop

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

import pandas as pd

# Создание DataFrame
data = {'Имя': ['Анна', 'Боб', 'Кейт'],
        'Возраст': [25, 30, 35],
        'Город': ['Москва', 'Санкт-Петербург', 'Екатеринбург']}
df = pd.DataFrame(data)

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

# Вывод измененного DataFrame
print(df)

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

Если вам нужно удалить колонку в pandas dataframe, вы можете использовать метод drop(). Этот метод позволяет удалять как одну колонку, так и несколько колонок.

Вот синтаксис метода drop():

dataframe.drop(labels, axis=1, inplace=True)

Разберем каждый параметр по отдельности:

• labels: Это параметр, который указывает на название или список названий колонок для удаления. Вместо этого параметра можно также использовать параметр columns с теми же значениями.
• axis: Параметр, определяющий, что мы хотим удалить - колонку или строку. В данном случае мы указываем axis=1, чтобы удалить колонку. Если вы хотите удалить строку, установите axis=0.
• inplace: Параметр, указывающий, хотим ли мы изменить исходный dataframe без создания нового. Установите inplace=True, чтобы изменения были применены к исходному dataframe, или inplace=False, чтобы создать новый dataframe с удаленной колонкой.

Вот примеры, показывающие, как удалить колонку или несколько колонок из dataframe:

Пример 1: Удаление одной колонки

import pandas as pd

# Создаем dataframe
data = {'Имя': ['Алексей', 'Егор', 'Мария', 'Ольга'],
        'Возраст': [25, 30, 28, 35],
        'Город': ['Москва', 'Санкт-Петербург', 'Екатеринбург', 'Новосибирск']}

df = pd.DataFrame(data)

# Удаляем колонку "Город"
df.drop('Город', axis=1, inplace=True)

print(df)

Вывод:

       Имя  Возраст
0  Алексей       25
1     Егор       30
2    Мария       28
3    Ольга       35

Пример 2: Удаление нескольких колонок

import pandas as pd

# Создаем dataframe
data = {'Имя': ['Алексей', 'Егор', 'Мария', 'Ольга'],
        'Возраст': [25, 30, 28, 35],
        'Город': ['Москва', 'Санкт-Петербург', 'Екатеринбург', 'Новосибирск'],
        'Профессия': ['Студент', 'Инженер', 'Менеджер', 'Врач']}

df = pd.DataFrame(data)

# Удаляем колонки "Город" и "Профессия"
df.drop(['Город', 'Профессия'], axis=1, inplace=True)

print(df)

Вывод:

       Имя  Возраст
0  Алексей       25
1     Егор       30
2    Мария       28
3    Ольга       35

Надеюсь, эта статья помогла вам понять, как удалить колонку в pandas dataframe. Используйте метод drop() с параметром axis=1 для удаления колонок. Удачи в работе с pandas!