为什么第二个for循环只迭代一次？

Question

import csv
with open("t1.csv", "rb") as f:
    reader = csv.reader(f)
    with open("t2.csv", "rb") as e:
         reader2 = csv.reader(e)
         with open("alnew.csv", "wb") as cr:
             writer = csv.writer(cr)
             for row in reader:
                 print row + ["XXXXX"]
                 for row2 in reader2:
                      print row2 + ["OOOOO"]
                      if row[0] == row2[3]:
                          new = row + row2
                          writer.writerow(new)

难道第二个for循环每次经过第一个循环就不应该迭代吗？ 它似乎只在它第一次通过循环时运行。

Answer 1

问题是您只打开一次文件句柄到第二个文件。 第二次通过循环已经耗尽。 像其他生成器一样，文件句柄只能迭代一次，而无需重新创建它们。 您可以每次重新打开文件。 但是，在这种情况下，我更喜欢itertools.product 。

import csv, itertools
with open("t1.csv", "rb") as f, with open("t2.csv", "rb") as e, with open("alnew.csv", "wb") as cr:
    reader = csv.reader(f)
    reader2 = csv.reader(e)
    writer = csv.writer(cr)
    for row, row2 in itertools.product(reader, reader2)
        if row[0] == row2[3]:
            writer.writerow(row + row2)

Answer 2

您需要“重置”阅读器。 在编写时，reader2将到达文件的末尾，但不知道它应该回到顶部。

要重新读取文件，您可以重新打开文件并每次重新开始。

import csv
with open("t1.csv", "rb") as f:
    reader = csv.reader(f)
    with open("alnew.csv", "wb") as cr:
        writer = csv.writer(cr)
        for row in reader:
            print row + ["XXXXX"]
            with open("t2.csv", "rb") as e:
                reader2 = csv.reader(e)
                for row2 in reader2:
                     print row2 + ["OOOOO"]
                     if row[0] == row2[3]:
                         new = row + row2
                         writer.writerow(new)

Answer 3

一旦完全阅读t2.csv ，阅读器（和基础文件）将保留在文件“末尾”。 它们不会仅仅因为您尝试再次遍历它们而重新设置为开始。

对于一个普通的文件，你可以使用seek方法来解决这个问题，但（除非我失去了一些东西） csv.reader没有seek方法，如果你调用可能不会很乐意与您seek在基础文件。 因此，最简单的解决方法是将整个t2.csv读入一个数组：change

     reader2 = csv.reader(e)

至

     reader2 = [row for row in csv.reader(e)]

如果t2.csv太大，这是现实，我认为你必须两个呼叫seek(0)上的文件，并摧毁和重建reader2 ，通过外环每次。