给定一个大文件(数百 MB),我将如何使用 Python 快速读取文件中特定开始和结束索引之间的内容?
本质上,我正在寻找一种更有效的方法:
open(filename).read()[start_index:end_index]
您可以seek
将文件放入文件中,然后从那里读取一定数量的文件。Seek 允许您获取文件中的特定偏移量,然后您可以将读取限制为仅该范围内的字节数。
with open(filename) as fin:
fin.seek(start_index)
data = fin.read(end_index - start_index)
那只会读取您正在寻找的数据。
这是我使用可变宽度编码的解决方案。我的 CSV 文件包含一个字典,其中每一行都是一个新项目。
def get_stuff(filename, count, start_index):
with open(filename, 'r') as infile:
reader = csv.reader(infile)
num = 0
for idx, row in enumerate(reader):
if idx >= start_index-1:
if num >= count:
return
else:
yield row
num += 1