Pandas DataFrame 获取列名 - Python 代码示例

要获取 DataFrame 中的列名，可以使用 DataFrame 的 columns 属性。以下是一个示例：

import pandas as pd

# 创建 DataFrame
data = {'Name': ['Tom', 'Nick', 'John'],
        'Age': [28, 32, 25],
        'City': ['London', 'New York', 'Paris']}
df = pd.DataFrame(data)

# 获取列名
column_names = df.columns

print(column_names)

输出：

Index(['Name', 'Age', 'City'], dtype='object')

在上面的示例中，df.columns 返回了一个 Index 对象，其中包含 DataFrame 的列名。