🔧Как преобразовать датафрейм в список python?🐍

Чтобы преобразовать датафрейм в список в Python, вы можете использовать метод values.tolist(). Вот пример:

import pandas as pd

# Создание датафрейма
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})

# Преобразование в список
df_list = df.values.tolist()

print(df_list)

Этот код создает датафрейм с двумя столбцами A и B. Затем мы используем метод values.tolist() для преобразования датафрейма в список df_list. Наконец, мы выводим этот список.

Преобразование датафрейма в список является одной из распространенных задач при работе с данными в Python. В этой статье мы рассмотрим различные подходы для выполнения этой задачи.

Использование метода values

Первый подход, который мы рассмотрим - это использование метода values для преобразования датафрейма в массив NumPy, а затем использование метода tolist для преобразования массива в список.

import pandas as pd
import numpy as np

# Создание датафрейма
data = {'Имя': ['Алиса', 'Боб', 'Кэрол'],
        'Возраст': [25, 30, 35]}
df = pd.DataFrame(data)

# Преобразование датафрейма в список
my_list = df.values.tolist()

print(my_list)

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

[['Алиса', 25], ['Боб', 30], ['Кэрол', 35]]

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

Еще один подход для преобразования датафрейма в список - это использование метода to_dict. Этот метод преобразует датафрейм в словарь, где ключами являются названия столбцов, а значениями - списки значений из столбцов.

import pandas as pd

# Создание датафрейма
data = {'Имя': ['Алиса', 'Боб', 'Кэрол'],
        'Возраст': [25, 30, 35]}
df = pd.DataFrame(data)

# Преобразование датафрейма в список
my_dict = df.to_dict(orient='list')

print(my_dict)

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

{'Имя': ['Алиса', 'Боб', 'Кэрол'], 'Возраст': [25, 30, 35]}

Использование метода apply

Еще один подход для преобразования датафрейма в список - это использование метода apply, который применяет заданную функцию к каждому элементу данных и возвращает результат в виде списка.

import pandas as pd

# Создание датафрейма
data = {'Имя': ['Алиса', 'Боб', 'Кэрол'],
        'Возраст': [25, 30, 35]}
df = pd.DataFrame(data)

# Функция для преобразования строки в список
def convert_to_list(row):
    return list(row)

# Преобразование датафрейма в список
my_list = df.apply(convert_to_list, axis=1).tolist()

print(my_list)

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

[['Алиса', 25], ['Боб', 30], ['Кэрол', 35]]

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

Последний подход, который мы рассмотрим, - это использование метода iterrows, который возвращает итератор по парам (индекс, строка) из данных датафрейма. Мы можем преобразовать каждую строку в список и добавить ее в список результатов.

import pandas as pd

# Создание датафрейма
data = {'Имя': ['Алиса', 'Боб', 'Кэрол'],
        'Возраст': [25, 30, 35]}
df = pd.DataFrame(data)

# Преобразование датафрейма в список
my_list = []
for index, row in df.iterrows():
    my_list.append(list(row))

print(my_list)

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

[['Алиса', 25], ['Боб', 30], ['Кэрол', 35]]

Теперь у нас есть несколько способов преобразовать датафрейм в список в Python. Выберите подход, который лучше всего подходит для вашей задачи и начинайте работу с данными в нужном вам формате.