Python csv.reader:如何返回文件顶部?

Nop*_*ope 49 python csv

当我使用csv.reader移动文件时,如何返回文件顶部.如果我用普通文件这样做,我可以做一些像"file.seek(0)"这样的事情.csv模块有类似的东西吗?

提前谢谢;)

Fed*_*oni 73

您可以直接搜索该文件.例如:

>>> f = open("csv.txt")
>>> c = csv.reader(f)
>>> for row in c: print row
['1', '2', '3']
['4', '5', '6']
>>> f.seek(0)
>>> for row in c: print row   # again
['1', '2', '3']
['4', '5', '6']
Run Code Online (Sandbox Code Playgroud)

  • 这确实再次开始,但如果文件有标题行,它将返回列名称作为第一行的“值”。所以如果你使用 DictReader,你会得到这样的一行:“{'name': 'name', 'dob': 'dob', 'email': 'email}” (3认同)
  • 如果您只有对阅读器的引用而不是文件句柄,有什么办法可以做到这一点?我正在编写一个函数,该函数将读取器作为输入,扫描所有行,(理想情况下)然后在解析后将文件位置重置为eek(0)。理想情况下,我们不必将文件参数也传递给这个函数,函数签名很简单:“csv_aggregate(reader, agg_functions, reset_after_calc=True)”我们可以使用“reader._file.seek(0)”或其他东西吗相似的? (2认同)

Eva*_*ark 13

您仍然可以使用file.seek(0).例如,看看以下内容:

import csv
file_handle = open("somefile.csv", "r")
reader = csv.reader(file_handle)
# Do stuff with reader
file_handle.seek(0)
# Do more stuff with reader as it is back at the beginning now
Run Code Online (Sandbox Code Playgroud)

这应该工作,因为csv.reader正在使用相同的.

  • 尽管file_handle.seek(0)确实有效,但reader.line_num计数并没有被重置,所以如果你想稍后再使用它,你就会遇到问题.您当然可以在搜索(0)之后重复'reader = csv.reader(file_handle)',但如果您在读取器中的外部'for row:'循环为'reader.line_num',那么这将无法正常工作'不增加.如果你用for循环替换for循环,那就解决了这个问题,但是不太确定你做了什么来整理上次读取时的文件结束情况. (8认同)