🔧 Как добавить столбец в датафрейм Python: простая инструкция

import pandas as pd

# Создание датафрейма
df = pd.DataFrame({'Имя': ['Анна', 'Борис', 'Елена'],
                   'Возраст': [24, 32, 41]})

# Добавление столбца
df['Зарплата'] = [50000, 60000, 70000]

# Вывод датафрейма
print(df)
   

Как добавить столбец в датафрейм Python

Добавление нового столбца в датафрейм является важной операцией в анализе данных с использованием Python. В этой статье мы рассмотрим различные способы добавления столбца в датафрейм с помощью библиотеки Pandas.

1. Добавление столбца с помощью операции присваивания

Самым простым способом добавления нового столбца в DataFrame является использование операции присваивания с индексом нового столбца. Давайте рассмотрим пример:

import pandas as pd

# Создание датафрейма
df = pd.DataFrame({'Имя': ['Алиса', 'Боб', 'Карл'], 'Возраст': [25, 30, 35]})

# Добавление нового столбца
df['Зарплата'] = [50000, 60000, 70000]

print(df)
    

В этом примере мы создаем датафрейм с двумя столбцами: 'Имя' и 'Возраст'. Затем мы добавляем новый столбец 'Зарплата' с помощью операции присваивания и списка значений. Результатом будет следующий датафрейм:

    Имя   Возраст  Зарплата
0  Алиса      25     50000
1    Боб      30     60000
2   Карл      35     70000
    

2. Добавление столбца с помощью метода insert()

Еще одним способом добавления столбца в датафрейм является использование метода insert(). Этот метод позволяет вставить новый столбец на определенную позицию в датафрейме. Давайте рассмотрим пример:

import pandas as pd

# Создание датафрейма
df = pd.DataFrame({'Имя': ['Алиса', 'Боб', 'Карл'], 'Возраст': [25, 30, 35]})

# Добавление нового столбца на позицию 1
df.insert(1, 'Зарплата', [50000, 60000, 70000])

print(df)
    

В этом примере мы создаем датафрейм с двумя столбцами: 'Имя' и 'Возраст'. Затем мы используем метод insert(), чтобы вставить новый столбец 'Зарплата' на позицию 1. Результатом будет следующий датафрейм:

    Имя  Зарплата  Возраст
0  Алиса     50000      25
1    Боб     60000      30
2   Карл     70000      35
    

3. Добавление столбца с помощью метода assign()

Третьим способом добавления нового столбца является использование метода assign(). Этот метод создает копию датафрейма с добавленным столбцом и возвращает его. Давайте рассмотрим пример:

import pandas as pd

# Создание датафрейма
df = pd.DataFrame({'Имя': ['Алиса', 'Боб', 'Карл'], 'Возраст': [25, 30, 35]})

# Добавление нового столбца с помощью метода assign()
df_with_salary = df.assign(Зарплата=[50000, 60000, 70000])

print(df_with_salary)
    

В этом примере мы создаем датафрейм с двумя столбцами: 'Имя' и 'Возраст'. Затем мы используем метод assign(), чтобы создать копию датафрейма с добавленным столбцом 'Зарплата'. Результатом будет следующий датафрейм:

    Имя  Возраст  Зарплата
0  Алиса      25     50000
1    Боб      30     60000
2   Карл      35     70000
    

4. Добавление столбца на основе вычислений

Кроме простого добавления нового столбца с фиксированными значениями, вы также можете добавить столбец на основе вычислений существующих столбцов. Давайте рассмотрим пример:

import pandas as pd

# Создание датафрейма
df = pd.DataFrame({'Имя': ['Алиса', 'Боб', 'Карл'], 'Возраст': [25, 30, 35]})

# Добавление нового столбца на основе вычислений
df['Год рождения'] = pd.datetime.today().year - df['Возраст']

print(df)
    

В этом примере мы создаем датафрейм с двумя столбцами: 'Имя' и 'Возраст'. Затем мы добавляем новый столбец 'Год рождения', основываясь на текущем годе минус значение столбца 'Возраст'. Результатом будет следующий датафрейм:

    Имя  Возраст  Год рождения
0  Алиса      25         1997
1    Боб      30         1992
2   Карл      35         1987
    

Заключение

В этой статье мы рассмотрели различные способы добавления столбца в датафрейм с использованием библиотеки Pandas в Python. Вы можете использовать операцию присваивания, метод insert(), метод assign() или добавить столбец на основе вычислений. Выберите подходящий способ в зависимости от вашего конкретного случая использования.