从 Pandas DataFrame 中检索列标题

Pandas DataFrame 是通用的数据结构，可以实现高效的数据操作和分析。一项常见任务涉及提取列标题，这对于获取 DataFrame 结构的概述或进一步处理非常有用。

假设您有一个从用户输入导入的 DataFrame，其中列的数量和名称未知。要将列标题提取为列表，您可以利用以下方法：

方法 1：使用 DataFrame.columns.values

column_headers = list(my_dataframe.columns.values)

my_dataframe.columns 返回包含列标题的 Index 对象。通过使用值将此索引转换为列表，您将获得表示列名称的字符串列表。

方法 2：使用 DataFrame.columns

column_headers = list(my_dataframe)

该方法是my_dataframe.columns.values的简写形式，直接将

用法示例

考虑 DataFrame:

    y  gdp  cap
0   1    2    5
1   2    3    9
2   8    7    2
3   3    4    7
4   6    7    7
5   4    8    3
6   8    2    8
7   9    9   10
8   6    6    4
9  10   10    7

使用任一方法，您将获得以下列标题列表：

['y', 'gdp', 'cap']