07 Июл 2023
2 мин
3652

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

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

Содержание

При работе с данными на языке программирования 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 можно несколькими способами. Рассмотрим наиболее часто используемые из них.

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

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

Содержание

Добавить комментарий

Определи профессию по рисунку