从Pandas DataFrame中获取列标题列表

获取Pandas DataFrame中的列标题列表可以通过以下3种方式实现：

1. 使用columns属性获取列标题列表

可以使用DataFrame的columns属性获取DataFrame的列标题列表，它返回一个Index对象：

import pandas as pd

df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]})

columns_list = df.columns.to_list()

print(columns_list)

输出结果：

['A', 'B', 'C']

可以使用DataFrame的keys方法获取DataFrame的列标题列表，它返回一个Index对象：

import pandas as pd

df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]})

columns_list = list(df.keys())

print(columns_list)

输出结果：

['A', 'B', 'C']

可以使用DataFrame的columns属性获取DataFrame的列标题列表，并调用tolist方法将其转换为列表：

import pandas as pd

df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]})

columns_list = df.columns.tolist()

print(columns_list)

输出结果：

['A', 'B', 'C']

以上三种方法均可用于获取Pandas DataFrame中的列标题列表。您可以根据自己的需求来选择使用哪种方法。

Tags: pandas