Python CSV 阅读器没有任何数据

Question

下面的代码工作正常。

with open(filename, 'rb') as f:
    reader = csv.reader(f)
    for row in reader:
        cur.execute(insertStatement, row)

当我插入这两行时，出了点问题。

with open(filename, 'rb') as f:
    reader = csv.reader(f)
    totalrows = len(list(reader))
    print totalrows # Print out the correct output
    for row in reader:
        cur.execute(insertStatement, row)   

我的猜测是，当我分配 totalrows = len(list(reader))光标移动到文件末尾，因为 for 循环中没有任何反应。

如果这是真的，我如何在不关闭文件并重新打开文件的情况下将光标移回开头？如果没有，请帮助。

Answer 1

没错，文件已经被消费，后续在reader上读取不会返回数据。

您可以通过在底层文件上调用 f.seek(0) 来解决这个问题，即

with open(filename, 'rb') as f:
    reader = csv.reader(f)
    totalrows = len(list(reader))
    print totalrows # Print out the correct output
    f.seek(0)
    for row in reader:
        cur.execute(insertStatement, row)