🔍 Как добавить столбец в dataframe python? Простая инструкция для начинающих!

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

Чтобы добавить столбец в dataframe в Python, можно воспользоваться методом df['Новый столбец'] = значения, где df - ваш dataframe, 'Новый столбец' - название нового столбца, а значения - данные, которыми хотите заполнить столбец.

Вот пример:

import pandas as pd

# Создаем dataframe
data = {'Имя': ['Анна', 'Иван', 'Мария'],
        'Возраст': [25, 32, 28]}
df = pd.DataFrame(data)

# Добавляем новый столбец
df['Работа'] = ['Программист', 'Аналитик', 'Дизайнер']

print(df)

В этом примере мы создаем dataframe с данными об именах и возрасте людей, а затем добавляем столбец 'Работа' со значениями 'Программист', 'Аналитик' и 'Дизайнер'.

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

В Python существует несколько способов добавить столбец в dataframe. Здесь мы рассмотрим несколько методов с примерами кода для каждого из них.

1. Используя метод assign()

Метод assign() является одним из наиболее простых способов добавить столбец в dataframe. Он позволяет создавать новый столбец, присваивая ему определенное значение или результат вычисления.

import pandas as pd

# Создаем dataframe
df = pd.DataFrame({'A': [1, 2, 3, 4],
                   'B': ['a', 'b', 'c', 'd']})

# Используем метод assign() для создания нового столбца
df = df.assign(C=[5, 6, 7, 8])

print(df)

Вывод:
A B C
0 1 a 5
1 2 b 6
2 3 c 7
3 4 d 8

2. Используя квадратные скобки

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

import pandas as pd

# Создаем dataframe
df = pd.DataFrame({'A': [1, 2, 3, 4],
                   'B': ['a', 'b', 'c', 'd']})

# Добавляем новый столбец
df['C'] = [5, 6, 7, 8]

print(df)

Вывод:
A B C
0 1 a 5
1 2 b 6
2 3 c 7
3 4 d 8

3. Используя метод insert()

Метод insert() позволяет добавить новый столбец в заданную позицию в dataframe. Вы указываете позицию и название столбца, а также значения для этого столбца.

import pandas as pd

# Создаем dataframe
df = pd.DataFrame({'A': [1, 2, 3, 4],
                   'B': ['a', 'b', 'c', 'd']})

# Используем метод insert() для добавления нового столбца
df.insert(1, 'C', [5, 6, 7, 8])

print(df)

Вывод:
A C B
0 1 5 a
1 2 6 b
2 3 7 c
3 4 8 d

В этом примере новый столбец 'C' вставляется в позицию 1, перед столбцом 'B'.

4. Используя операторы присваивания

Один из простых способов добавить столбец в dataframe - использовать операторы присваивания, такие как = или +=.

import pandas as pd

# Создаем dataframe
df = pd.DataFrame({'A': [1, 2, 3, 4],
                   'B': ['a', 'b', 'c', 'd']})

# Добавляем новый столбец с использованием оператора присваивания
df['C'] = 5

print(df)

Вывод:
A B C
0 1 a 5
1 2 b 5
2 3 c 5
3 4 d 5

В этом примере новый столбец 'C' получает значение 5 для всех строк dataframe.

Заключение

В этой статье мы рассмотрели несколько способов добавить столбец в dataframe Python. Метод assign(), использование квадратных скобок, метод insert() и операторы присваивания - все они позволяют легко добавить новый столбец с помощью кода.