python - 写入和读取 ndarray

Question

我想保存一个形状为 (5,2) 的数组，该数组名为 sorted_cube_station_list。

在打印中它看起来不错，但是当我用 numpy.tofile 保存它然后用 numpy.fromfile 读取它时它变成了一个一维数组

你能帮我吗？ 将 numpy 导入为 num

nx=5
ny=5
nz=5
stations=['L001','L002','L003','L004','L005']

for x in range(nx):
    for y in range (ny):
         for z in range (nz):
             cube_station_list = []
             i=-1
             for sta in stations:
                 i=i+1
                 cube=[int(i), num.random.randint(2500, size=1)[0]]
                 cube_station_list.append(cube)
             cub_station_list_arr=num.asarray(cube_station_list)
             sorted_cube_station_list_arr=cub_station_list_arr[cub_station_list_arr[:, 1].argsort()]
             print x,y,z, sorted_cube_station_list_arr
             num.ndarray.tofile(sorted_cube_station_list_arr,str(x)+'_'+str(y)+'_'+str(z)

Answer 1

我建议你使用np.save

a = np.ones(16).reshape([8, 2])
np.save("fileName.npy", a)

参见 the docs : 第一个参数一定不能是你要保存的变量，而是你要保存的文件路径。因此，您在使用 np.save(yourArray)

时遇到的错误

您可以使用np.load(pathToArray)加载保存的数组