numpy数组存入文件与取出文件以及注意事项(tofile和fromfile)
我们知道numpy的array是可以保存到文件的,一个常用的做法是通过to_file()保存到而进行.bin文件中,然后再通过from_file()从.bin文件中将其读取出来,下面看一个例子。
如下图所示spatiallyRepresentation
是一个二维numpy
数组,其shape
为[31762, 8]
接下来将其存入文件中,使用tofile
方法即可,参数填入想要保存到的文件路径,然后使用fromfile
可以将其从文件中读取出来。
但是可以发现,读取出来的test
的shape
变成1
维了,更细心点会发现test
的最前面几个值和之前看到的spatiallyRepresentation
的值也不一样啊,这是为什么呢?
首先,使用tofile
方法,会默认将所有数据按顺序排成一个向量,然后以二进制形式存入文件中,而读取的时候自然会变成1维了,如果已知原始数组的维数,将它reshape
一下就行了。
然而即便是reshape
后,test
也不等于spatiallyRepresentation
,即我读取出来的和存进去的数据不一样!
这个解决办法很简单,这是因为读取数据的时候需要指定数据类型。如第一张图里看到的,原始数据是float32类型的。
所以按下图方法读取即可