从Pandas DataFrame中获取列标题列表

在这篇文章中，我们将看到，如何在Python中以列表形式获得Pandas数据框架的所有列标题。

DataFrame.column.values属性将返回一个列标题的数组。

pandas DataFrame列名

使用list()在Pandas数据框架中以列表形式获取列名

在这个方法中，我们使用Python内置的list()函数，即list(df.columns.values)，函数。

# import pandas library
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]})
 
print("The DataFrame :")
display(df.head())
 
# multiple ways of getting column names as list
print("\nThe column headers :")
print("Column headers from list(df.columns.values):",
      list(df.columns.values))
print("Column headers from list(df):", list(df))
print("Column headers from list(df.columns):",
      list(df.columns))

输出 :

从Pandas DataFrame中获取列标题列表

使用list()从pandas DataFrame中获取列列表

注意：这里我们有display()函数，它在Jupyter笔记本内工作，用于演示。如果要在其他IDE中运行，你可以用print()函数代替display()函数。

使用tolist()在Pandas数据框架中以列表形式获取列名

在这个方法中，我们正在导入Python pandas模块并创建一个DataFrame，以获得列表中的列名，我们正在使用tolist() , 函数。

# import pandas library
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()
print("The column headers :")
 
print(df.columns.tolist())
# or we could also use
# print(df.columns.values.tolist())
# or,
# print(df.columns.to_numpy().tolist())

输出 :

The DataFrame :
  Students   BMI   Religion
0        A  22.7      Hindu
1        B  18.0      Islam
2        C  21.4  Christian
3        D  24.1       Sikh
The column headers :
['Students', 'BMI', 'Religion']

使用列表理解在Pandas数据框架中以列表形式获取列名

在这个方法中，我们正在导入一个Pandas模块并创建一个Dataframe来获取列表中的列名，我们正在使用列表理解。

# 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])

输出:

从Pandas DataFrame中获取列标题列表