我有十行CSV文件。从这个文件中,我只想要第四行。最快的方法是什么?我正在寻找类似的东西:
with open(file, 'r') as my_file:
reader = csv.reader(my_file)
print reader[3]
其中reader[3]
显然是我想要实现的语法不正确。如何将阅读器移至第4行并获取其内容?
答案 0 :(得分:5)
如果您只有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))