解决csv.reader迭代器只能迭代一次的问题

在做项目过程中,需要对同一个CSV文件进行两次迭代,但是在第二次迭代时候,输出为空。

这里用一个小例子表示:

解决csv.reader迭代器只能迭代一次的问题

对这个简单的csv文件迭代第一次:

解决csv.reader迭代器只能迭代一次的问题

迭代第二次:

解决csv.reader迭代器只能迭代一次的问题

可以看到,第二次迭代的时候,输出为空。

检查下迭代器可以发现,迭代器已经指到文件末尾了。这也是为什么第二次输出为空的原因。

解决csv.reader迭代器只能迭代一次的问题

因此需要对迭代器进行刷新。

暴力方式:生成新的迭代器(注意这里的迭代器指的是csv.reader,而不是item_t,因此如果只是将item_t改成如item_tt之类的是不起作用的。)

解决csv.reader迭代器只能迭代一次的问题

在网上还看到其他方法:(可以看看https://blog.****.net/m0_37321987/article/details/105727403

解决csv.reader迭代器只能迭代一次的问题

但是这个弊端也比较大,相当于建了个列表,把CSV文件放进去,这样就能遍历任意多次,但这个对内存也是很大的负担。

具体看https://cloud.tencent.com/developer/article/1534178

虽然讲的是生成器的,但是原理是一样的。

当然,如果要遍历一个csv文件不止一次,且可能需要多次访问,最保险的也是最方便的方式还是生产列表或者dataframe等将csv内容存下来。