无法迭代" csv.reader"重复 - 第二次迭代时清空结果

时间:2018-01-01 12:22:55

标签: python python-3.x list csv nested-loops

我已创建list1以包含特定列中的唯一值,方法是附加标识为唯一的值,以便执行某种"总和如果"。

下面是我正在努力的场景,内部循环,迭代csv行只执行一次。是否有一些特殊的特性使得在嵌套循环的内部循环中迭代csv文件中的行有问题?

csv内容:

Field1
row1
row2
row3

代码:

datafile = open(r"my_file.csv", "r")
myreader = csv.reader(datafile, delimiter=",")

list1 = ["A", "B", "C"]

for x in list1[:]:
    print(x)
        for y in myreader:
               print(y)

预期结果:

A
row1
row2
row3
B
row1
row2
row3
C
row1
row2
row3

实际结果:

A
row1
row2
row3<
B
C

似乎只有内循环的初始迭代才有效。

如果我用其他列表替换csv,它可以正常工作:

datafile = open(r"my_file.csv", "r")
myreader = csv.reader(datafile, delimiter=",")

list1 = ["A", "B", "C"]
list2 = ["row1", "row2", "row2"]

for x in list2[:]:
    print(x)
        for y in list2:
               print(y)

给出预期结果

我可以在单个循环中迭代csv行,下面没有问题:

x = 0
for y in myreader:
        x = x + 1
        print(y[2] + " INNER")

那么为什么一起工作呢?

3 个答案:

答案 0 :(得分:2)

csv.reader返回一个reader对象,它是一个迭代器。因此,一旦迭代整个列表,它就会变空。您需要将阅读器保存为list,以便重复进行迭代。

为了实现这一目标,您需要更改:

myreader = list(csv.reader(datafile, delimiter=","))
#           ^ it'll save `myreader` as list

因此,您完成的代码应为:

datafile = open(r"my_file.csv", "r")
myreader = list(csv.reader(datafile, delimiter=","))    
#           ^ converting the `iterator` to list type

list1 = ["A", "B", "C"]

#            v  I removed "[:]" from here because it is not required
#            v  `list1[:]` creates a copy of the list "list1". You 
#            v  don't need copy here
for x in list1:
    print(x)
    for y in myreader:
        print(y)

答案 1 :(得分:1)

在使用A进行循环后,您的阅读器迭代器将显示所有数据。 您必须重新定位它,但为什么要这样做?

最好将所有数据保存到列表中,然后重复使用。

myData = list(myreader) # this will iterate the whole file and put all in a list.

请参阅https://docs.python.org/3/library/csv.html

答案 2 :(得分:1)

myreader是一个迭代器,在第一次迭代后会耗尽。 将其转换为列表以获得所需的输出。

csv_rows = list(myreader)
for x in list1[:]:
    print(x)
        for y in csv_rows:
               print(y)

迭代器只允许一次迭代。另一方面,列表可以反复迭代。迭代器的优点是不会立即将所有元素加载到内存中。如果文件很大,这可能很重要。在这种情况下,列表可能会占用大量内存。