您好, 欢迎来到 !    登录 | 注册 | | 设为首页 | 收藏本站

Python生成器读取大型CSV文件

Python生成器读取大型CSV文件

您可以具有一个生成器,该生成器从两个不同的csv阅读器读取行并将其行作为数组对产生。该代码是:

import csv
import numpy as np

def getData(filename1, filename2):
    with open(filename1, "rb") as csv1, open(filename2, "rb") as csv2:
        reader1 = csv.reader(csv1)
        reader2 = csv.reader(csv2)
        for row1, row2 in zip(reader1, reader2):
            yield (np.array(row1, dtype=np.float),
                   np.array(row2, dtype=np.float)) 
                # This will give arrays of floats, for other types change dtype

for tup in getData("file1", "file2"):
    print(tup)
python 2022/1/1 18:49:36 有373人围观

撰写回答


你尚未登录,登录后可以

和开发者交流问题的细节

关注并接收问题和回答的更新提醒

参与内容的编辑和改进,让解决方法与时俱进

请先登录

推荐问题


联系我
置顶