使用CSV对象列出理解行为

Question

我正在尝试从Python 2.6中的CSV对象构造两个简单列表。在我的所有测试中，第二个生成的列表总是返回为空（第一个生成的列表返回正常。）这几乎就像第二个列表推导在第一个完成之前开始。我可以让这段代码按预期运行的唯一方法是将CSV数据复制到自己的列表中，然后迭代副本 - 这很好，但似乎效率低下。

我很难过。

可疑代码：

import csv

data_source = '/charts.csv'
final_data = []
x_obs1 = []
y_obs1 = []
x_obs2 = []
y_obs2 = []

data_file = open(data_source, "r")
final_data = csv.reader(data_file, delimiter=',')

[(x_obs1.append(item[0]), y_obs1.append(item[1])) for item in final_data]
[(x_obs2.append(item[0]), y_obs2.append(item[2])) for item in final_data]

data_file.close()

print y_obs2

返回：[]

这可以按预期工作：

import csv

data_source = '/charts.csv'
final_data = []
x_obs1 = []
y_obs1 = []
x_obs2 = []
y_obs2 = []

data_file = open(data_source, "r")
csv_data = csv.reader(data_file, delimiter=',')

[final_data.append(item) for item in csv_data]

data_file.close()

[(x_obs1.append(item[0]), y_obs1.append(item[1])) for item in final_data]
[(x_obs2.append(item[0]), y_obs2.append(item[2])) for item in final_data]

print y_obs2

返回：['Barometric Pressure Normal', '30', '30', '30', '30', '30'....

我错过了什么？任何见解都会受到最高的赞赏。

Answer 1

基本上，csv.reader仅适用于csv文件。第二次尝试迭代同一csv.reader时，它已经用完了，所以返回一个空列表。

所以你的第二种情况是更正确的做你需要的方式。它创建了CSV数据的缓存副本，而无需返回磁盘上的文件（这相对较贵）。