⚡️Как добавить столбец в сводную таблицу с помощью Питона?💪

import pandas as pd

# Создаем исходный DataFrame
data = {'Город': ['Москва', 'Санкт-Петербург', 'Москва', 'Санкт-Петербург'],
        'Погода': ['Солнечно', 'Облачно', 'Дождь', 'Солнечно'],
        'Температура': [25, 20, 18, 23]}

df = pd.DataFrame(data)

# Создаем сводную таблицу с новым столбцом "Влажность"
pivot_table = pd.pivot_table(df, values='Температура', index='Город', columns='Погода', aggfunc='mean', fill_value='-')

# Добавляем новый столбец "Влажность"
pivot_table['Влажность'] = [65, 70]

print(pivot_table)
    

Как добавить столбец в сводную таблицу в Python

Сводная таблица (Pivot table) - это мощный инструмент для анализа данных, который позволяет суммировать и агрегировать информацию на основе определенных столбцов. Чтобы добавить новый столбец в сводную таблицу в Python, мы можем использовать функцию pandas.DataFrame.assign(). Давайте рассмотрим подробнее, как это сделать.

Шаг 1: Импорт библиотек

Первым шагом необходимо импортировать необходимые библиотеки для работы с данными:

import pandas as pd

Шаг 2: Создание сводной таблицы

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

data = {
    'Город': ['Москва', 'Санкт-Петербург', 'Москва', 'Санкт-Петербург'],
    'Категория': ['Автомобили', 'Автомобили', 'Недвижимость', 'Недвижимость'],
    'Цена': [500000, 600000, 7000000, 8000000]
}

df = pd.DataFrame(data)

Теперь мы можем создать сводную таблицу с помощью функции pandas.pivot_table():

pivot_table = pd.pivot_table(df, values='Цена', index='Город', columns='Категория')

Шаг 3: Добавление нового столбца

Теперь, чтобы добавить новый столбец в сводную таблицу, мы можем использовать функцию pandas.DataFrame.assign(). В качестве аргумента передаем имя нового столбца и задаем ему значение:

pivot_table = pivot_table.assign(Средняя_Цена=round(pivot_table.mean(axis=1), 2))

В приведенном выше примере мы добавляем новый столбец с именем "Средняя_Цена", который содержит среднюю цену по каждому городу.

Шаг 4: Полный пример кода

Вот полный пример кода, который показывает, как добавить столбец в сводную таблицу:

import pandas as pd

data = {
    'Город': ['Москва', 'Санкт-Петербург', 'Москва', 'Санкт-Петербург'],
    'Категория': ['Автомобили', 'Автомобили', 'Недвижимость', 'Недвижимость'],
    'Цена': [500000, 600000, 7000000, 8000000]
}

df = pd.DataFrame(data)

pivot_table = pd.pivot_table(df, values='Цена', index='Город', columns='Категория')

pivot_table = pivot_table.assign(Средняя_Цена=round(pivot_table.mean(axis=1), 2))

print(pivot_table)

Результат выполнения этого кода будет выглядеть следующим образом:

Категория         Автомобили  Недвижимость  Средняя_Цена
Город                                                
Москва                500000       7000000    3750000.00
Санкт-Петербург       600000       8000000    4300000.00

Таким образом, нам удалось успешно добавить новый столбец "Средняя_Цена" в сводную таблицу.