[人民日报] 如何将 Pandas DataFrame 列标题获取为列表[央视网]

[人民日报] 如何将 Pandas DataFrame 列标题获取为列表[央视网]

已邀请:

fish - Hadooper

Pandas 是用于 Python 数据分析的开源软件包。pandas.DataFrame 是主要的 Pandas 数据结构。它是带有标签轴(行和列)的二维表格数据结构。
我们来介绍如何从 DataFrame 对象获取列标题的列表。
示例中使用相同的如下面定义的 DataFrame 对象。
>>> import pandas
>>> cities = {
... 'name': ['New York', 'Los Angeles', 'Chicago'],
... 'population': [8601186, 4057841, 2679044],
... 'state': ['NY', 'CA', 'IL'],
... }
>>> data_frame = pandas.DataFrame(cities)
获取 DataFrame 列名的一种方法是迭代 DataFrame 对象本身。DataFrame 迭代器按定义顺序返回列名。
>>> for column in data_frame:
... print(column)
...
name
population
state
当需要将可迭代对象转换为列表时,可以在其上调用 Python 的内置 list 函数。
>>> list(data_frame)
['name', 'population', 'state']
我们还可以从 DataFrame.columns 属性访问其列。

>>> list(data_frame.columns)
['name', 'population', 'state']
否则,我们可以使用 DataFrame.columns.tolist() 函数来实现相同的目的。

>>> data_frame.columns.tolist()
['name', 'population', 'state']

遍历 DataFrame.columns.values 属性时,事情发生了很大变化。同样,与 DataFrame 对象和 DataFrame.columns 属性一样,我们可以使用它来获取一系列 DataFrame 列名。

还可以使用内置 DataFrame.columns.values.tolist() 方法
>>> data_frame.columns.values.tolist()
['name', 'population', 'state']

要回复问题请先登录注册