Obtenga una lista de los valores de una columna en particular de un Pandas DataFrame

En este artículo, veremos cómo obtener todos los valores de una columna en un marco de datos de pandas en forma de lista. Esto puede ser muy útil en muchas situaciones, supongamos que tenemos que sacar notas de todos los alumnos de una materia en particular, sacar números de teléfono de todos los empleados, etc. Veamos cómo podemos lograr esto con la ayuda de algunos ejemplos.

Ejemplo 1: Podemos tener todos los valores de una columna en una lista, usando el método tolist() .

Sintaxis: Serie.tolist().

Tipo de retorno: Serie convertida en Lista.

Código :

# import pandas libraey
import pandas as pd
  
# dictionary
dict = {'Name': ['Martha', 'Tim',
                'Rob', 'Georgia'],
         'Marks': [87, 91, 
                  97, 95]}
  
# create a dataframe object
df = pd.DataFrame(dict)
  
# show the dataframe
print(df)
  
# list of values of 'Marks' column
marks_list = df['Marks'].tolist()
  
# show the list
print(marks_list)

Producción:

Ejemplo 2: Veremos cómo podemos obtener los valores de todas las columnas en listas separadas.

Código:

# import pandas library
import pandas as pd
  
# dictionary
dict = {'Name': ['Martha', 'Tim', 
                'Rob', 'Georgia'],
        'Marks': [87, 91, 
                 97, 95]}
  
# create a dataframe object
df = pd.DataFrame(dict)
  
# show the dataframe
print(df)
  
# iterating over and calling 
# tolist() method for 
# each column
for i in list(df):
    
    # show the list of values  
    print(df[i].tolist())

Producción: