如何在 Pandas 数据框中获取列名

在 Pandas 中，我们通常使用数据框（DataFrame）来存储和操作数据。数据框由行和列组成，每一列都有一个名称，该名称通常称为列名。

在许多情况下，我们需要获取数据框中的列名，以便进行操作。下面，我们将介绍如何在 Pandas 数据框中获取列名。

方法1：columns 属性

Pandas 数据框对象有一个 columns 属性，该属性返回一个包含所有列名的列表。以下是一个示例：

import pandas as pd

data = {'name': ['Alice', 'Bob', 'Charlie', 'David'],
        'age': [25, 32, 18, 47],
        'gender': ['F', 'M', 'M', 'M']}
df = pd.DataFrame(data)

print(df.columns)

输出结果为：

Index(['name', 'age', 'gender'], dtype='object')

我们可以看到，df.columns 属性返回了一个包含所有列名的列表。

方法2：keys() 方法

Pandas 数据框对象还有一个 keys() 方法，该方法也可以用于获取列名。以下是一个示例：

import pandas as pd

data = {'name': ['Alice', 'Bob', 'Charlie', 'David'],
        'age': [25, 32, 18, 47],
        'gender': ['F', 'M', 'M', 'M']}
df = pd.DataFrame(data)

print(df.keys())

输出结果为：

Index(['name', 'age', 'gender'], dtype='object')

我们可以看到，df.keys() 方法也返回了一个包含所有列名的列表。

需要注意的是，df.columns 属性和 df.keys() 方法的运行结果相同，但它们是不同的对象。

小结

在 Pandas 中，我们可以使用 df.columns 属性或 df.keys() 方法获取数据框中的列名。这两种方法都返回一个包含所有列名的列表。需要注意的是，df.columns 属性和 df.keys() 方法虽然返回的结果相同，但它们是不同的对象。