🔥 Как добавить столбец в таблицу pandas быстро и легко?

  import pandas as pd

  # Создание таблицы
  df = pd.DataFrame({'A': [1, 2, 3],
                     'B': [4, 5, 6]})
  
  # Добавление нового столбца "C"
  df = df.assign(C=[7, 8, 9])

  # Вывод таблицы с новым столбцом
  print(df)
  

Как добавить столбец в таблицу pandas

Добавление столбца в таблицу pandas - важная операция, позволяющая расширить функциональность и обогатить данные. Ниже приведен подробный обзор, объясняющий, как это сделать.

Использование метода assign()

Один из способов добавления столбца в таблицу pandas - использование метода assign(). Этот метод позволяет создавать новый столбец на основе существующих данных и присваивать его таблице.

import pandas as pd

# Создание исходного DataFrame
df = pd.DataFrame({'Name': ['John', 'Alice', 'Bob'],
                   'Age': [25, 30, 35]})

# Добавление нового столбца с использованием метода assign()
df = df.assign(City=['New York', 'London', 'Paris'])

print(df)

В этом примере мы создаем исходный DataFrame с двумя столбцами ("Name" и "Age"). Затем мы используем метод assign(), чтобы добавить новый столбец с названием "City" и значениями ["New York", "London", "Paris"]. После присвоения мы выводим обновленный DataFrame с помощью функции print().

Использование оператора присваивания

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

import pandas as pd

# Создание исходного DataFrame
df = pd.DataFrame({'Name': ['John', 'Alice', 'Bob'],
                   'Age': [25, 30, 35]})

# Добавление нового столбца с использованием оператора присваивания
df['City'] = ['New York', 'London', 'Paris']

print(df)

В этом примере мы также создаем исходный DataFrame с двумя столбцами ("Name" и "Age"). Затем мы используем оператор присваивания, чтобы добавить новый столбец с названием "City" и значениями ["New York", "London", "Paris"]. После присвоения мы выводим обновленный DataFrame с помощью функции print().

Использование функции loc()

Функция loc() - еще один способ добавления столбца в таблицу pandas. Этот метод позволяет непосредственно обращаться к таблице и присваивать значения новому столбцу.

import pandas as pd

# Создание исходного DataFrame
df = pd.DataFrame({'Name': ['John', 'Alice', 'Bob'],
                   'Age': [25, 30, 35]})

# Добавление нового столбца с использованием функции loc()
df.loc[:, 'City'] = ['New York', 'London', 'Paris']

print(df)

В этом примере мы создаем исходный DataFrame с двумя столбцами ("Name" и "Age"). Затем мы используем функцию loc() с двумя параметрами: срезом по всем строкам (:) и названием нового столбца ("City"). Мы присваиваем новые значения ["New York", "London", "Paris"] этой комбинации среза и столбца. После присвоения мы выводим обновленный DataFrame с помощью функции print().

Заключение

Добавление столбца в таблицу pandas - важный шаг для расширения функциональности данных. В этой статье мы рассмотрели три способа добавления столбца: использование метода assign(), оператора присваивания и функции loc(). Каждый из них предоставляет удобные возможности для работы с данными.