1. Introduction
En Python, un dataframe es una estructura de datos bidimensional que se utiliza para almacenar y manipular datos tabulares. A veces, es posible que necesitemos convertir un dataframe en una lista para realizar ciertas operaciones o análisis de datos. En este tutorial, exploraremos diferentes formas de convertir un dataframe en una lista en Python.
2. Converting a dataframe into a list using the values attribute
Una forma sencilla de convertir un dataframe en una lista es utilizando el atributo «values» del dataframe. El atributo «values» devuelve una representación de matriz de los datos en el dataframe. Podemos utilizar este atributo para convertir el dataframe en una lista de listas, donde cada lista interna representa una fila del dataframe.
Aquí hay un ejemplo de cómo convertir un dataframe en una lista utilizando el atributo «values»:
«`python
import pandas as pd
# Crear un dataframe de ejemplo
data = {‘Nombre’: [‘Juan’, ‘María’, ‘Pedro’],
‘Edad’: [25, 30, 35],
‘Ciudad’: [‘Madrid’, ‘Barcelona’, ‘Valencia’]}
df = pd.DataFrame(data)
# Convertir el dataframe en una lista
lista = df.values.tolist()
print(lista)
«`
El resultado será:
«`
[[‘Juan’, 25, ‘Madrid’], [‘María’, 30, ‘Barcelona’], [‘Pedro’, 35, ‘Valencia’]]
«`
En este ejemplo, creamos un dataframe con tres columnas: «Nombre», «Edad» y «Ciudad». Luego, utilizamos el atributo «values» para convertir el dataframe en una lista de listas.
3. Converting a dataframe into a list using the tolist() method
Otra forma de convertir un dataframe en una lista es utilizando el método «tolist()» proporcionado por la biblioteca pandas. El método «tolist()» devuelve una representación de lista de los datos en el dataframe.
Aquí hay un ejemplo de cómo convertir un dataframe en una lista utilizando el método «tolist()»:
«`python
import pandas as pd
# Crear un dataframe de ejemplo
data = {‘Nombre’: [‘Juan’, ‘María’, ‘Pedro’],
‘Edad’: [25, 30, 35],
‘Ciudad’: [‘Madrid’, ‘Barcelona’, ‘Valencia’]}
df = pd.DataFrame(data)
# Convertir el dataframe en una lista
lista = df.tolist()
print(lista)
«`
El resultado será:
«`
[[‘Juan’, 25, ‘Madrid’], [‘María’, 30, ‘Barcelona’], [‘Pedro’, 35, ‘Valencia’]]
«`
En este ejemplo, creamos un dataframe con tres columnas: «Nombre», «Edad» y «Ciudad». Luego, utilizamos el método «tolist()» para convertir el dataframe en una lista.
4. Converting a specific column of a dataframe into a list
A veces, es posible que solo necesitemos convertir una columna específica de un dataframe en una lista. Podemos lograr esto utilizando la notación de corchetes para acceder a la columna deseada y luego aplicando el método «tolist()».
Aquí hay un ejemplo de cómo convertir una columna específica de un dataframe en una lista:
«`python
import pandas as pd
# Crear un dataframe de ejemplo
data = {‘Nombre’: [‘Juan’, ‘María’, ‘Pedro’],
‘Edad’: [25, 30, 35],
‘Ciudad’: [‘Madrid’, ‘Barcelona’, ‘Valencia’]}
df = pd.DataFrame(data)
# Convertir la columna ‘Nombre’ en una lista
lista_nombre = df[‘Nombre’].tolist()
print(lista_nombre)
«`
El resultado será:
«`
[‘Juan’, ‘María’, ‘Pedro’]
«`
En este ejemplo, creamos un dataframe con tres columnas: «Nombre», «Edad» y «Ciudad». Luego, utilizamos la notación de corchetes para acceder a la columna ‘Nombre’ y aplicamos el método «tolist()» para convertir esa columna en una lista.
5. Converting multiple columns of a dataframe into a list
Si queremos convertir múltiples columnas de un dataframe en una lista, podemos utilizar la notación de corchetes para acceder a las columnas deseadas y luego aplicar el método «values.tolist()».
Aquí hay un ejemplo de cómo convertir múltiples columnas de un dataframe en una lista:
«`python
import pandas as pd
# Crear un dataframe de ejemplo
data = {‘Nombre’: [‘Juan’, ‘María’, ‘Pedro’],
‘Edad’: [25, 30, 35],
‘Ciudad’: [‘Madrid’, ‘Barcelona’, ‘Valencia’]}
df = pd.DataFrame(data)
# Convertir las columnas ‘Nombre’ y ‘Edad’ en una lista
lista_columnas = df[[‘Nombre’, ‘Edad’]].values.tolist()
print(lista_columnas)
«`
El resultado será:
«`
[[‘Juan’, 25], [‘María’, 30], [‘Pedro’, 35]]
«`
En este ejemplo, creamos un dataframe con tres columnas: «Nombre», «Edad» y «Ciudad». Luego, utilizamos la notación de corchetes para acceder a las columnas ‘Nombre’ y ‘Edad’ y aplicamos el método «values.tolist()» para convertir esas columnas en una lista.
6. Conclusion
En este tutorial, hemos explorado diferentes formas de convertir un dataframe en una lista en Python. Hemos aprendido a utilizar el atributo «values» y el método «tolist()» para lograr esto. También hemos visto cómo convertir una columna específica o múltiples columnas de un dataframe en una lista. Espero que este tutorial te haya sido útil y te ayude a trabajar con dataframes y listas en Python.
