如何从 Pandas DataFrame 中提取列标题？

从 Pandas DataFrame 中检索列标题

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

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

方法 1：使用 DataFrame.columns.values

<code class="python">column_headers = list(my_dataframe.columns.values)</code>

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

方法 2：使用 DataFrame.columns

<code class="python">column_headers = list(my_dataframe)</code>

This method 是 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']

以上是如何从 Pandas DataFrame 中提取列标题？的详细内容。更多信息请关注PHP中文网其他相关文章！