python取定义的列名的数据

在Python中，要取定义的列名的数据，通常情况下需要使用pandas库。下面介绍两种常用的获取数据的方法：

方法一：使用pandas的read_excel()函数读取Excel文件，然后使用DataFrame的列名获取数据。示例如下：

import pandas as pd

# 读取Excel文件
df = pd.read_excel("data.xlsx")

# 获取指定列的数据
col_data = df["列名"].values.tolist()

方法二：使用pandas的read_csv()函数读取csv文件，设置参数header=0来指定第一行为列名，然后使用DataFrame的列名获取数据。示例如下：

import pandas as pd

# 读取csv文件
df = pd.read_csv("data.csv", header=0)

# 获取指定列的数据
col_data = df["列名"].values.tolist()

其中，tolist()函数用于将pandas的Series转换为Python列表。除了读取文件，如果已经有DataFrame对象，也可以直接使用列名获取数据，示例如下：

import pandas as pd

# 创建DataFrame对象
data = {"列名1": [1, 2, 3], "列名2": [4, 5, 6]}
df = pd.DataFrame(data)

# 获取指定列的数据
col_data = df["列名1"].values.tolist()