从 Pandas DataFrame 中获取列标题列表

在使用 Pandas 进行数据操作时，获取 DataFrame 中的列标题列表是一项基本操作。本文将介绍如何从 Pandas DataFrame 中获取列标题列表。

方法一：使用 columns 属性

Pandas DataFrame 对象包含一个名为 columns 的属性，这个属性是一个 Index 对象，可以使用 tolist() 方法将其转换为列表类型。

import pandas as pd

# 创建一个 DataFrame
data = {'name': ['Alice', 'Bob', 'Charlie'], 
        'age': [25, 30, 35], 
        'gender': ['female', 'male', 'male']}

df = pd.DataFrame(data)

# 使用 columns 属性获取列标题列表
columns_list = df.columns.tolist()

print(columns_list)
# 输出：['name', 'age', 'gender']

方法二：使用 keys() 方法

Pandas DataFrame 对象还有一个名为 keys() 的方法，它返回的结果是一个类似于字典的键列表，也可以使用 tolist() 方法将其转换为列表类型。

import pandas as pd

# 创建一个 DataFrame
data = {'name': ['Alice', 'Bob', 'Charlie'], 
        'age': [25, 30, 35], 
        'gender': ['female', 'male', 'male']}

df = pd.DataFrame(data)

# 使用 keys() 方法获取列标题列表
columns_list = df.keys().tolist()

print(columns_list)
# 输出：['name', 'age', 'gender']

无论是使用 columns 属性还是 keys() 方法，获取 Pandas DataFrame 中的列标题列表都非常简单。同时，如果需要获取 DataFrame 中的行标题列表，可以使用 index 属性或者类似于 keys() 方法的 index 方法。