Как удалить столбец в dataframe python 🗑️

    import pandas as pd

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

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

    print(df)
    

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

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

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

Первый способ удаления столбца - это использование метода drop(). Метод drop() принимает в качестве аргумента имя столбца, который нужно удалить.

import pandas as pd

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

df = pd.DataFrame(data)

# Удаление столбца 'Возраст'
df.drop('Возраст', axis=1, inplace=True)

print(df)

В результате выполнения кода будет удален столбец 'Возраст' из DataFrame.

2. Удаление столбца по индексу

Второй способ удаления столбца - это использование индекса столбца. Для этого мы можем использовать атрибут iloc[]. Атрибут iloc[] принимает в качестве аргумента индекс столбца, который нужно удалить.

import pandas as pd

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

df = pd.DataFrame(data)

# Удаление столбца с индексом 1
df.drop(df.columns[1], axis=1, inplace=True)

print(df)

В результате выполнения кода будет удален столбец с индексом 1 (в данном случае столбец 'Возраст') из DataFrame.

3. Удаление столбца с условием

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

import pandas as pd

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

df = pd.DataFrame(data)

# Удаление столбца, если его имя содержит 'Возраст'
df = df.loc[:, ~df.columns.str.contains('Возраст')]

print(df)

В результате выполнения кода будет удален столбец с именем, содержащим 'Возраст' из DataFrame.

4. Удаление нескольких столбцов

Четвертый способ - удаление нескольких столбцов одновременно. Для этого мы можем передать список имен столбцов, которые нужно удалить, в метод drop().

import pandas as pd

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

df = pd.DataFrame(data)

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

print(df)

В результате выполнения кода будут удалены столбцы 'Возраст' и 'Город' из DataFrame.

Заключение

В этой статье мы рассмотрели различные способы удаления столбца в DataFrame с использованием языка Python. Вы можете выбрать подходящий метод в зависимости от ваших конкретных требований.