При работе с данными на языке программирования 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. Для её решения существуют различные инструменты и методы, которые позволяют выполнять данную операцию быстро и эффективно.
Добавить комментарий