Как преобразовать DataFrame Pandas в список?

Опубликовано: 27 Марта, 2022

Let’s discuss how to convert Pandas dataframe to List. First, let’s create a Basic DataFrame:

Python3

import pandas as pd 
 
# Creating a dictionary to store data
data = {"Name":["Tony", "Steve", "Bruce", "Peter" ],
        "Age": [35, 70, 45, 20] } 
 
# Creating DataFrame 
df = pd.DataFrame(data) 
 
# Print the dataframe
df

Выход :

Иногда вам может потребоваться преобразовать фрейм данных pandas в List. Для выполнения этой задачи можно использовать функцию tolist (). Ниже приведен базовый пример использования этой функции и преобразования необходимого DataFrame в список.

Python3

df.values.tolist()

Выход :

[[«Тони», 35], [«Стив», 70], [«Брюс», 45], [«Питер», 20]]

Здесь каждый внутренний список содержит все столбцы определенной строки.

Pandas DataFrame можно преобразовать в списки несколькими способами. Давайте посмотрим на разные способы преобразования DataFrame один за другим.

Метод №1: преобразование DataFrame в список, содержащий все строки определенного столбца:

Python3

import pandas as pd 
 
# Creating a dictionary to store data
data = {"Name":["Tony", "Steve", "Bruce", "Peter" ] ,
        "Age": [35, 70, 45, 20] } 
 
# Creating DataFrame 
df = pd.DataFrame(data) 
 
# Converting DataFrame to a list containing
# all the rows of column "Name"
names = df["Name"].tolist()
 
# Printing the converted list.
print(names)

Выход:

[«Тони», «Стив», «Брюс», «Питер»]

Метод № 2: преобразование DataFrame во вложенный список, содержащий все строки всех столбцов:

Python3

import pandas as pd 
 
# Creating a dictionary to store data
data = {"Name":["Tony", "Steve", "Bruce", "Peter" ] ,
        "Age": [35, 70, 45, 20] } 
 
# Creating DataFrame
df = pd.DataFrame(data)
 
# Creating an empty list
res=[]
 
# Iterating through the columns of
# dataframe
for column in df.columns:
     
    # Storing the rows of a column
    # into a temporary list
    li = df[column].tolist()
     
    # appending the temporary list
    res.append(li)
     
# Printing the final list
print(res)

Выход:

[[«Тони», «Стив», «Брюс», «Питер»], [35, 70, 45, 20]]

Метод № 3: преобразование DataFrame в список, содержащий списки, содержащие все столбцы строки.

Python3

import pandas as pd 
 
# Creating a dictionary to store data
data = {"Name":["Tony", "Steve", "Bruce", "Peter" ] ,
        "Age": [35, 70, 45, 20] } 
 
# Creating DataFrame
df = pd.DataFrame(data) 
 
# Converting dataframe to list
li = df.values.tolist()
 
# Printing list
print(li)

Выход :

[[«Тони», 35], [«Стив», 70], [«Брюс», 45], [«Питер», 20]]

Метод №4: преобразование DataFrame в список, содержащий списки, содержащие все столбцы строки вместе с именами столбцов.

Python3

import pandas as pd 
 
# Creating a dictionary to store data
data = {"Name":["Tony", "Steve", "Bruce", "Peter" ] ,
        "Age": [35, 70, 45, 20] } 
 
# Creating DataFrame
df = pd.DataFrame(data) 
 
# Converting dataframe to list
li = [df.columns.values.tolist()] + df.values.tolist()
 
# Printing list
print(li)

Выход:

[[‘Name’, ‘Age’], [‘Tony’, 35], [‘Steve’, 70], [‘Bruce’, 45], [‘Peter’, 20]]

Внимание компьютерщик! Укрепите свои основы с помощью базового курса программирования Python и изучите основы.

Для начала подготовьтесь к собеседованию. Расширьте свои концепции структур данных с помощью курса Python DS. А чтобы начать свое путешествие по машинному обучению, присоединяйтесь к курсу Машинное обучение - базовый уровень.

Python

Как преобразовать DataFrame Pandas в список?

Python3

Python3

Python3

Python3

Python3

Python3

РЕКОМЕНДУЕМЫЕ СТАТЬИ