
如题,数据量比较大,想把一个 ndarray 保存到本地,这两种方法和直接 write 到本地性能相比怎么样,我试了下 savetxt()有点慢,请问大家有什么比较好的方法吗
1 glasslion 2018-11-26 17:20:47 +08:00 试试 Apache Arrow(Apache Parquet Format) |
2 xpresslink 2018-11-26 18:30:06 +08:00 cPickle 了解一下 |
3 DongDongXie OP @xpresslink 以前用过这个,据说比 pickle 快 |
4 deepreader 2018-11-27 11:47:00 +08:00 massage pack 了解一下 |
5 wwqgtxx 2018-11-27 15:22:36 +08:00 |
6 wwqgtxx 2018-11-27 15:48:44 +08:00 其实还有一些蹊跷意淫的办法,比如你先 pickle 保存 ndarray 的 dtype size shape 三个属性,然后用 memcpy 直接保存 ndarray 的 data,那是一个 memoryview 对象,你可以用 mmap 或者直接 file.write 保存进文件,读的时候用 np.frombuffer 或者 np.fromstring 读进来后 reshape 一下就好了( frombuffer 的话最后需要 copy 一下,否则当原 buffer 对象失效之后会出问题,当然如果你原 buffer 对象是从 mmap 获得的话那也没事,而且后续对该 ndarray 的修改还能用过 mmap.flush 写回硬盘) |