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

Методы преобразования DataFrame в список списков:

Метод 1: использование атрибута values
Пример кода:

import pandas as pd
# Create a sample DataFrame
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
# Convert DataFrame to a list of lists
list_of_lists = df.values.tolist()
print(list_of_lists)

Выход:

[[1, 4], [2, 5], [3, 6]]

Метод 2: использование понимания списка
Пример кода:

import pandas as pd
# Create a sample DataFrame
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
# Convert DataFrame to a list of lists
list_of_lists = [list(row) for row in df.values]
print(list_of_lists)

Выход:

[[1, 4], [2, 5], [3, 6]]

Метод 3: использование метода to_records()
Пример кода:

import pandas as pd
# Create a sample DataFrame
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
# Convert DataFrame to a list of lists
list_of_lists = [list(row)[1:] for row in df.to_records()]
print(list_of_lists)

Выход:

[[1, 4], [2, 5], [3, 6]]

Метод 4: использование метода iterrows()
Пример кода:

import pandas as pd
# Create a sample DataFrame
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
# Convert DataFrame to a list of lists
list_of_lists = [list(row[1]) for row in df.iterrows()]
print(list_of_lists)

Выход:

[[1, 4], [2, 5], [3, 6]]

Метод 5: использование метода apply()
Пример кода:

import pandas as pd
# Create a sample DataFrame
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
# Convert DataFrame to a list of lists
list_of_lists = df.apply(list, axis=1).tolist()
print(list_of_lists)

Выход:

[[1, 4], [2, 5], [3, 6]]

В этой статье мы рассмотрели несколько методов преобразования DataFrame в список списков с использованием библиотеки Python pandas. Мы обсудили пять различных методов: использование атрибута values, понимание списка, метод to_records(), метод iterrows()и метод apply()метод. Каждый метод обеспечивает простой способ достижения желаемого преобразования, обеспечивая гибкость в зависимости от конкретных требований. Используя эти методы, вы можете легко преобразовать данные DataFrame в формат, подходящий для дальнейшего анализа или интеграции с другими библиотеками Python.