1. 程式人生 > >numpy儲存資料

numpy儲存資料

儲存numpy array資料

利用numpy.save()函式將array儲存為.npy格式的資料

numpy中的資料I/O可以簡單地通過save函式實現:

import numpy as np

a = np.array(range(100))

np.save('where/you/wanto/store/output',a)    #numpy 會自動加上.npy字尾

這時在目錄中得到out.npy的檔案。
讀入的時候使用load()即可:

import numpy as np

b = np.load('here/you/wanto/store/output.npy'
) np.array_equal(a,b) #驗證 比較a,b是否相同時同一個array #Output is: True

numpy手冊save(),load()