如何在读取csv熊猫之前导入过滤的行

Question

嗨，我必须在pandas数据框中上传大量的csv文件。加载之前是否可以从这些csv文件中过滤掉数据，以免出现任何内存错误。

我现有的设置给我内存错误

我有一个位置列，其中包含32个值，但是我只希望在导入之前过滤3-4个位置。

这可能吗？

Answer 1

您可以使用csv library逐行读取并仅保留所需的记录：

import csv
with open('names.csv', newline='') as csvfile:
     reader = csv.DictReader(csvfile)
     for row in reader:
         print(row['first_name'], row['last_name'])

之后，您可以使用writerow

将过滤后的行保存到csv文件中