Как добавить строку в DataFrame pandas
Введение
Работа с DataFrame в pandas — это основа анализа данных в Python. DataFrame представляет собой двумерную таблицу данных, аналогичную таблицам в базах данных или электронных таблицах Excel. Одной из часто встречающихся задач является добавление новой строки в уже существующий DataFrame. В этой статье мы рассмотрим несколько способов, как это сделать, используя методы loc
и append
. Мы также обсудим, в каких ситуациях каждый из этих методов может быть наиболее полезен, и приведем примеры кода, которые помогут вам лучше понять их применение.
Создание DataFrame
Прежде чем мы начнем добавлять строки, давайте создадим примерный DataFrame. Это поможет нам лучше понять, как работают методы добавления строк. DataFrame можно создать из различных источников данных, таких как списки, словари или даже файлы CSV. В нашем примере мы будем использовать словарь для создания DataFrame.
import pandas as pd
# Создаем DataFrame
data = {
'Name': ['Alice', 'Bob', 'Charlie'],
'Age': [25, 30, 35],
'City': ['New York', 'Los Angeles', 'Chicago']
}
df = pd.DataFrame(data)
print(df)
Этот код создаст DataFrame с тремя столбцами: Name
, Age
и City
, и тремя строками. DataFrame будет выглядеть следующим образом:
Name Age City
0 Alice 25 New York
1 Bob 30 Los Angeles
2 Charlie 35 Chicago
Добавление строки с использованием loc
Метод loc
позволяет добавлять строки, указывая индекс новой строки и значения для каждого столбца. Этот метод особенно полезен, если вам нужно добавить строку в определенное место DataFrame. Например, если вы хотите добавить строку с индексом 3, вы можете использовать следующий код:
# Добавляем новую строку с использованием loc
df.loc[3] = ['David', 40, 'San Francisco']
print(df)
После выполнения этого кода, DataFrame будет выглядеть следующим образом:
Name Age City
0 Alice 25 New York
1 Bob 30 Los Angeles
2 Charlie 35 Chicago
3 David 40 San Francisco
Метод loc
также позволяет добавлять строки в любое место DataFrame, указывая соответствующий индекс. Например, если вы хотите добавить строку с индексом 1.5, вы можете использовать следующий код:
# Добавляем новую строку с использованием loc
df.loc[1\.5] = ['Eve', 28, 'Boston']
print(df)
После выполнения этого кода, DataFrame будет выглядеть следующим образом:
Name Age City
0 Alice 25 New York
1 Bob 30 Los Angeles
1.5 Eve 28 Boston
2 Charlie 35 Chicago
3 David 40 San Francisco
Добавление строки с использованием append
Метод append
позволяет добавлять строки к DataFrame, создавая новый DataFrame. Этот метод удобен, если вам нужно добавить несколько строк или если вы хотите сохранить оригинальный DataFrame неизменным. Например, если вы хотите добавить строку с именем Eve
, возрастом 28 и городом Boston
, вы можете использовать следующий код:
# Создаем новую строку в виде словаря
new_row = {'Name': 'Eve', 'Age': 28, 'City': 'Boston'}
# Добавляем новую строку с использованием append
df = df.append(new_row, ignore_index=True)
print(df)
После выполнения этого кода, DataFrame будет выглядеть следующим образом:
Name Age City
0 Alice 25 New York
1 Bob 30 Los Angeles
2 Charlie 35 Chicago
3 David 40 San Francisco
4 Eve 28 Boston
Метод append
также позволяет добавлять несколько строк одновременно. Например, если вы хотите добавить две новые строки, вы можете использовать следующий код:
# Создаем новые строки в виде списка словарей
new_rows = [
{'Name': 'Frank', 'Age': 33, 'City': 'Seattle'},
{'Name': 'Grace', 'Age': 29, 'City': 'Austin'}
]
# Добавляем новые строки с использованием append
df = df.append(new_rows, ignore_index=True)
print(df)
После выполнения этого кода, DataFrame будет выглядеть следующим образом:
Name Age City
0 Alice 25 New York
1 Bob 30 Los Angeles
2 Charlie 35 Chicago
3 David 40 San Francisco
4 Eve 28 Boston
5 Frank 33 Seattle
6 Grace 29 Austin
Заключение
Теперь вы знаете два основных способа добавления строк в DataFrame pandas: с использованием методов loc
и append
. Оба метода имеют свои преимущества и могут быть использованы в зависимости от конкретной задачи. Метод loc
позволяет добавлять строки в определенное место DataFrame и может быть полезен, если вам нужно добавить строку с определенным индексом. Метод append
позволяет добавлять строки, создавая новый DataFrame, и может быть удобен, если вам нужно добавить несколько строк или если вы хотите сохранить оригинальный DataFrame неизменным.
Надеемся, что эта статья помогла вам лучше понять, как работать с DataFrame в pandas. Теперь вы можете использовать эти методы для добавления строк в ваши собственные DataFrame и улучшить свои навыки анализа данных в Python.