Veamos cómo obtener todos los encabezados de columna de un Pandas DataFrame como una lista en Python .
El atributo df.columns.values devolverá una lista de encabezados de columna.
Método 1: Obtener nombres de columna como lista en Pandas DataFrame usando list()
En este método, estamos importando un módulo de Pandas y creando un marco de datos para obtener los nombres de las columnas en una lista, estamos usando la función list(df.columns.values) .
Python3
# importing pandas as pd
import pandas as pd
# creating the dataframe
df = pd.DataFrame({'PassengerId': [892, 893, 894, 895,
896, 897, 898, 899],
'PassengerClass': [1, 1, 2, 1, 3, 3, 2, 2],
'PassengerName': ['John', 'Prity', 'Harry',
'Smith', 'James', 'Amora',
'Kiara', 'Joseph'],
'Age': [32, 54, 71, 21, 37, 9, 11, 54]})
display("The DataFrame :")
display(df)
# print the list of all the column headers
display("The column headers :")
display(list(df.columns.values))
Producción :
Método 2: Obtener nombres de columna como lista en Pandas DataFrame usando tolist()
En este método estamos importando un módulo de Pandas y creando un marco de datos para obtener los nombres de las columnas en una lista que estamos usando la función tolist() .
Python3
# importing pandas as pd
import pandas as pd
# creating the dataframe
my_df = {'Students': ['A', 'B', 'C', 'D'],
'BMI': [22.7, 18.0, 21.4, 24.1],
'Religion': ['Hindu', 'Islam',
'Christian', 'Sikh']}
df = pd.DataFrame(my_df)
display("The DataFrame :")
display(df)
# print the list using tolist()
display("The column headers :")
display(df.columns.values.tolist())
Producción :
Método 3: Obtener nombres de columna como lista en Pandas DataFrame usando la comprensión de lista
En este método, estamos importando un módulo de Pandas y creando un marco de datos para obtener los nombres de las columnas en una lista, estamos usando la comprensión de listas .
Python3
# importing pandas as pd
import pandas as pd
# creating the dataframe
my_df = {'Students': ['A', 'B', 'C', 'D'],
'BMI': [22.7, 18.0, 21.4, 24.1],
'Religion': ['Hindu', 'Islam',
'Christian', 'Sikh']}
df = pd.DataFrame(my_df)
display("The DataFrame :")
display(df)
# print the list of all the column headers
display("The column headers :")
# display using list comprehension
print([col for col in df])
Producción: