Pandas: Преобразование DataFrame в список словарей

Pandas: Преобразование DataFrame в список словарей

Введение

Pandas – это мощная библиотека для анализа данных, которая предоставляет удобные инструменты для работы с таблицами и временными рядами. Одним из распространенных заданий при работе с данными является преобразование DataFrame в список словарей. В этой статье мы рассмотрим различные способы выполнения этой задачи с использованием библиотеки Pandas.

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

Метод to_dict() – это один из простых способов преобразования DataFrame в список словарей. При вызове этого метода без аргументов он возвращает словарь, в котором ключами являются названия столбцов DataFrame, а значениями – соответствующие элементы каждого столбца.

import pandas as pd

# Создаем DataFrame
data = {'Name': ['John', 'Emma', 'Kevin'],
        'Age': [28, 32, 45],
        'City': ['New York', 'London', 'Paris']}
df = pd.DataFrame(data)

# Преобразуем DataFrame в список словарей
dict_list = df.to_dict(orient='records')
print(dict_list)

Вывод:

[{'Name': 'John', 'Age': 28, 'City': 'New York'},
 {'Name': 'Emma', 'Age': 32, 'City': 'London'},
 {'Name': 'Kevin', 'Age': 45, 'City': 'Paris'}]

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

Еще один способ преобразования DataFrame в список словарей – использование метода iterrows(). Этот метод позволяет итерироваться по строкам DataFrame и создавать словарь для каждой строки.

import pandas as pd

# Создаем DataFrame
data = {'Name': ['John', 'Emma', 'Kevin'],
        'Age': [28, 32, 45],
        'City': ['New York', 'London', 'Paris']}
df = pd.DataFrame(data)

# Преобразуем DataFrame в список словарей
dict_list = []
for index, row in df.iterrows():
    dict_list.append(row.to_dict())
print(dict_list)

Вывод:

[{'Name': 'John', 'Age': 28, 'City': 'New York'},
 {'Name': 'Emma', 'Age': 32, 'City': 'London'},
 {'Name': 'Kevin', 'Age': 45, 'City': 'Paris'}]

Заключение

Преобразование DataFrame в список словарей является распространенной задачей при работе с библиотекой Pandas. В этой статье мы рассмотрели два простых способа выполнения этой задачи с помощью методов to_dict() и iterrows().

Благодаря такому преобразованию, мы можем легко работать с данными в формате словарей, который может быть удобным для дальнейшей обработки и анализа. Мы надеемся, что эта статья помогла вам лучше понять процесс преобразования DataFrame в список словарей с помощью библиотеки Pandas.

Читайте так же  Получение всех аргументов, переданных в функцию Python

Ссылки