Добавление нового столбца в существующий DataFrame в Python

При работе с данными на языке программирования Python с использованием библиотеки pandas часто возникает задача добавления нового столбца в существующий DataFrame. DataFrame — это двумерная маркированная структура данных с возможностью хранения разнородных типов данных. Пример DataFrame представлен ниже:

import pandas as pd
data = {'a': [1, 2, 3], 'b': [4, 5, 6], 'c': [7, 8, 9]}
df = pd.DataFrame(data)
print(df)

Вывод:

   a  b  c
0  1  4  7
1  2  5  8
2  3  6  9

Добавить новый столбец в DataFrame можно несколькими способами. Рассмотрим наиболее часто используемые из них.

Освойте Python на курсе от Skypro. Вас ждут 400 часов обучения и практики (достаточно десяти часов в неделю), подготовка проектов для портфолио, индивидуальная проверка домашних заданий и помощь опытных наставников. Получится, даже если у вас нет опыта в IT.

1. Добавление столбца путем присвоения данных новому столбцу

Этот метод заключается в прямом присвоении данных новому столбцу DataFrame. Например, если необходимо добавить столбец ‘d’, содержащий числа от 10 до 12, то это можно сделать следующим образом:

df['d'] = [10, 11, 12]
print(df)

Вывод:

   a  b  c   d
0  1  4  7  10
1  2  5  8  11
2  3  6  9  12

2. Использование функции assign

Функция assign позволяет добавлять новый столбец, используя синтаксис функций. Например, добавим столбец ‘e’, который является суммой столбцов ‘a’ и ‘b’:

df = df.assign(e = df['a'] + df['b'])
print(df)

Вывод:

   a  b  c   d   e
0  1  4  7  10   5
1  2  5  8  11   7
2  3  6  9  12   9

Изучайте Python на онлайн-курсе от Skypro «Python-разработчик». Программа рассчитана на новичков без опыта программирования и технического образования. Курс проходит в формате записанных коротких видеолекций. Будет много проверочных заданий и мастер-классов. В конце каждой недели — живая встреча с экспертами в разработке для ответов на вопросы и разбора домашек.

Курсы по программированию

Помогаем получить новую профессию с гарантией трудоустройства!

Подробнее

3. Использование метода insert

Метод insert позволяет добавить столбец на определенную позицию в DataFrame. Например, добавим столбец ‘f’ на вторую позицию:

df.insert(1, 'f', [13, 14, 15])
print(df)

Вывод:

   a   f  b  c   d   e
0  1  13  4  7  10   5
1  2  14  5  8  11   7
2  3  15  6  9  12   9

В заключение, добавление новых столбцов в DataFrame — это часто встречающаяся задача при обработке данных в Python. Для её решения существуют различные инструменты и методы, которые позволяют выполнять данную операцию быстро и эффективно.

Тест на профориентацию

За 10 минут узнайте, как ваш опыт пригодиться в IT индустрии

Подробнее