¿Cómo convertir Pandas DataFrame en una lista?

Analicemos cómo convertir el marco de datos de Pandas a List. Primero, creemos un marco de datos básico:

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

Producción :

A veces, es posible que deba convertir su marco de datos de pandas a List. Para realizar esta tarea, se puede utilizar la función ‘ tolist() ‘. A continuación se muestra un ejemplo básico para usar esta función y convertir el DataFrame requerido en una lista.

df.values.tolist()

Producción : 

[['Tony', 35], ['Steve', 70], ['Bruce', 45], ['Peter', 20]]

Aquí, cada lista interna contiene todas las columnas de una fila en particular. 

Pandas DataFrame se puede convertir en listas de varias maneras. Echemos un vistazo a las diferentes formas de convertir un DataFrame uno por uno.

Método n.º 1: convertir un marco de datos en una lista que contenga todas las filas de una columna en particular:

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)

Producción: 

['Tony', 'Steve', 'Bruce', 'Peter']

Método n.º 2: convertir un marco de datos en una lista anidada que contenga todas las filas de todas las columnas:

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)

Producción: 

[['Tony', 'Steve', 'Bruce', 'Peter'], [35, 70, 45, 20]]

Método #3: Convertir un DataFrame en una lista que contiene listas que tienen todas las columnas de una fila.

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)

Producción :

[['Tony', 35], ['Steve', 70], ['Bruce', 45], ['Peter', 20]]

Método #4: Convertir un DataFrame en una lista que contiene listas que tienen todas las columnas de una fila junto con los nombres de las columnas.

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)

Producción:

[[‘Nombre’, ‘Edad’], [‘Tony’, 35], [‘Steve’, 70], [‘Bruce’, 45], [‘Peter’, 20]]