如何使用Pandas框架将CSV文件中的列数据存入数组

没问题，这事儿超级好搞定，直接在你已有的代码基础上补充两行就行。Pandas的DataFrame列可以轻松转换成Python列表，用tolist()方法就完事了：

import pandas as pd
df = pd.read_csv('things.csv')
things = []
date = []

# 把CSV列数据转成列表存入对应变量
things = df['thing'].tolist()
date = df['date'].tolist()

简单解释下： #

• df['thing'] 会取出CSV文件里名为thing的那一列（注意这里的列名必须和你CSV里的列名完全一致，大小写也不能错）
• .tolist() 方法会把Pandas的Series对象（也就是这一列的数据）转换成标准的Python列表，直接赋值给你预先定义好的things和date变量就OK了。

小提示： #

如果你的CSV里有缺失值，想要过滤掉空值再转列表的话，可以加个dropna()：

things = df['thing'].dropna().tolist()
date = df['date'].dropna().tolist()

举个例子，假设你的things.csv内容是这样的：

thing,date
笔记本,2024-05-01
钢笔,2024-05-02
便签本,2024-05-03

运行代码后，things会变成['笔记本', '钢笔', '便签本']，date则是['2024-05-01', '2024-05-02', '2024-05-03']，完全符合你的需求～

内容的提问来源于stack exchange，提问作者SvyatoslavKvartovkinVEVO