Python-读取大量表格数据的有效方法

时间:2019-09-16 17:21:12

标签: python optimization

我有一个文件,其中包含一个大数字表,大小约为300 MB。我想用Python阅读。

数据如下:

-200 1 11097.4 16414.2 1
-200 1 11197.4 16414.8 1
-200 1 11297.4 16415.4 1
-200 1 11397.4 16416 1
-200 1 11497.4 16416.5 1
-200 1 11597.4 16417.1 1
-200 1 11697.4 16417.7 1

Python代码如下:

    with open(filename) as f:
        nrow, ncol= [int(x) for x in next(f).split()] 
        for k in range(2):
            rr = []
            for i in range(nrow+1):
                row = []
                for j in range(ncol+1):
                    a = next(f).split()                     
                    row.append([int(a[0]), int(a[1]), float(a[2]), float(a[4])])
                rr.append(row)          
            summary.append(rr)

这非常慢;读取文件大约需要60秒钟。我想把时间降到10秒以内。使其更快一点的最简单方法是什么?

如果有帮助,我很高兴更改数据文件格式。

1 个答案:

答案 0 :(得分:3)

使用熊猫。这可能是重复的,所以也请查看这些答案

code.py

import pandas as pd
import numpy as np

df = pd.read_csv("large_file.txt", sep="\s")
np.save("large_file.npz", df.values)

with load('large_file.npz') as data:
    print(data.shape)