python - 是否可以打开一个大的 csv 而不将其完全加载到 RAM

Question

假设我有一个非常大的 csv 文件。

file = open("foo.csv")

似乎将整个 csv 放入 RAM 中。如果我只需要 csv 的第一行，但不希望 python 加载打开的整个文件，有没有办法做到这一点？

Answer 1

如果您只需要第一行，那么您可以像这样使用 csv 模块。

import csv

with open("foo.csv", "r") as my_csv:
    reader = csv.reader(my_csv)

    first_row = next(reader)

# do stuff with first_row

CSV 模块使用生成器，因此整个文件不会加载到 RAM 中，而是根据请求加载更多行，以防止整个文件加载到 RAM 中。