python - 从 CSV 文件中读取特定行

Question

我有一个十行的 CSV 文件。从这个文件中，我只想要第四行。最快的方法是什么？我正在寻找类似的东西:

with open(file, 'r') as my_file:
    reader = csv.reader(my_file)
    print reader[3]

reader[3] 显然是我想要实现的不正确语法。如何将阅读器移动到第 4 行并获取其内容？

Answer 1

如果只有 10 行，则可以将整个文件加载到列表中:

with open(file, 'r') as my_file:
    reader = csv.reader(my_file)
    rows = list(reader)
    print rows[3]

对于更大的文件，使用 itertools.islice() :

from itertools import islice

with open(file, 'r') as my_file:
    reader = csv.reader(my_file)
    print next(islice(reader, 3, 4))