我已创建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")
那么为什么一起工作呢?
答案 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.
答案 2 :(得分:1)
myreader
是一个迭代器,在第一次迭代后会耗尽。
将其转换为列表以获得所需的输出。
csv_rows = list(myreader)
for x in list1[:]:
print(x)
for y in csv_rows:
print(y)
迭代器只允许一次迭代。另一方面,列表可以反复迭代。迭代器的优点是不会立即将所有元素加载到内存中。如果文件很大,这可能很重要。在这种情况下,列表可能会占用大量内存。