Question

我正在尝试读取csv文件的行。我的文件看起来像这样

Col 1, Col 2, Col3
row11, row12, row13
row21, row22, row23
row31, row32, row33
...

我使用以下命令读取行

with open('~/data.csv') as f:
    r = csv.DictReader(f)
    for i in range(5):
        print(list(r)[i])

输出打印第一行，但随后它会立即给出索引错误。

IndexError                                Traceback (most recent call last)
<ipython-input-15-efcc4f8c760d> in <module>()
      2     r = csv.DictReader(f)
      3     for i in range(5):
----> 4         print(list(r)[i])

IndexError: list index out of range

我猜我在某个地方犯了一个愚蠢的错误，但无法发现它。关于我做错了什么以及如何解决它的任何想法？

编辑：这是print(list(r))的输出：

[{'Col 1': 'row11', ' Col3': ' row13', ' Col 2': ' row12'}, {'Col 1': 'row21', ' Col3': ' row23', ' Col 2': ' row22'}, {'Col 1': 'row31', ' Col3': ' row33', ' Col 2': ' row32'}, {'Col 1': 'row41', ' Col3': ' row43', ' Col 2': ' row42'}, {'Col 1': 'row51', ' Col3': ' row53', ' Col 2': ' row52'}, {'Col 1': 'row61', ' Col3': ' row63', ' Col 2': ' row62'}, {'Col 1': 'row71', ' Col3': ' row73', ' Col 2': ' row72'}, {'Col 1': 'row81', ' Col3': ' row83', ' Col 2': ' row82'}, {'Col 1': 'row91', ' Col3': ' row93', ' Col 2': ' row92'}, {'Col 1': 'row101', ' Col3': ' row103', ' Col 2': ' row102'}]

Answer 1

DictReader(f)只是让您一次性查看您的文件 - 您只能在其上调用list一次，但您可以多次调用它，因为它在循环内。以后的调用返回空list。只需在循环外部调用list并将其保存在变量中，您就会变成金色。

那是：

r = csv.DictReader(f)
rows = list(r)
for i in range(5):
    print(rows[i])

或者，不要在任何时候将整个事情都记入记忆中：

for row in csv.DictReader(f):
    print row

如果您希望将索引保留用于其他目的：

 for i, row in enumerate(csv.DictReader(f)):
     print i, row

如果您希望从iterator获取特定行（csv.DictReader是一个特殊情况）而不将整个内容拉入内存，请在{{3}处查看itertools.islice }。它基本上允许在list上进行iterator样式切片。

  # prints first five rows
  for row in itertools.islice(csv.DictReader(f), 5):
       print row

对于更多零星的行：

  needed_row_indices = {2, 5, 20}
  for i, row in enumerate(csv.DictReader(f)):
      if i in needed_row_indices:
          print row

我仍然在范围内得到IndexError

1 个答案: