python dataFrame columns 取得列表-CFANZ编程社区

Python DataFrame 列取得列表的实现指南

在数据分析领域，Pandas是一个非常强大的库，尤其是在处理表格数据时，更是不可或缺。今天我们就来学习如何从一个Pandas DataFrame 中获取列名，并将其转化为一个列表。以下是实现的具体流程和步骤。

流程概览

步骤	内容
1	导入必要的库
2	创建一个DataFrame
3	获取DataFrame的列名并转为列表
4	输出列名列表

1. 导入必要的库

首先，我们需要导入Pandas库。如果尚未安装，请使用pip install pandas进行安装。

import pandas as pd  # 引入 pandas 库

2. 创建一个DataFrame

接下来，我们创建一个简单的DataFrame以便于后续操作。可以利用字典创建DataFrame，其键为列名，值为列的数据。

data = {
    'Name': ['Alice', 'Bob', 'Charlie'],
    'Age': [25, 30, 35],
    'City': ['New York', 'Los Angeles', 'Chicago']
}

df = pd.DataFrame(data)  # 创建 DataFrame

3. 获取DataFrame的列名并转为列表

现在，我们要获取DataFrame的列名并将其转换为列表。Pandas提供了一个非常简单的方法来实现这一点。

columns_list = df.columns.tolist()  # 获取列名并转为列表

df.columns 返回的是一个索引对象，包含DataFrame的列名；
我们使用 .tolist() 方法将其转换为Python中的列表。

4. 输出列名列表

最后，我们可以输出这份列表，确认我们的操作是否成功。

print(columns_list)  # 输出列名列表

全部代码合并示例

通过上述步骤，我们总结上面的代码如下：

import pandas as pd  # 引入 pandas 库

# 创建 DataFrame
data = {
    'Name': ['Alice', 'Bob', 'Charlie'],
    'Age': [25, 30, 35],
    'City': ['New York', 'Los Angeles', 'Chicago']
}

df = pd.DataFrame(data)  # 创建 DataFrame

# 获取列名并转为列表
columns_list = df.columns.tolist()  # 获取列名并转为列表

# 输出列名列表
print(columns_list)  # 输出结果为：['Name', 'Age', 'City']

类图示例

在知识的传递过程中，了解各个组件之间的关系非常重要。下面是一个简单的类图，展示Pandas DataFrame的相关结构。

classDiagram
    class DataFrame {
        <<Pandas>>
        +columns
        +tolist()
    }
    class Series {
        <<Pandas>>
    }
    DataFrame --> Series : columns

在上面的类图中，我们可以看到DataFrame 类与Series 类之间的关系，DataFrame 类的 columns 属性返回的是一个Series。