在Python中保存数组有几种常见的方法,以下是使用`pickle`模块和`numpy`库保存数组的方法:
使用`pickle`模块
import pickle
创建一个数组
my_array = [1, 2, 3, 4, 5]
保存数组到文件
with open('array.pkl', 'wb') as file:
pickle.dump(my_array, file)
从文件加载数组
with open('array.pkl', 'rb') as file:
loaded_array = pickle.load(file)
print(loaded_array) 输出: [1, 2, 3, 4, 5]
使用`numpy`库
保存为`.npy`文件
import numpy as np
创建一个数组
a = np.array([1, 2, 3, 4, 5])
保存数组到文件
np.save('test00000001.npy', a)
从文件加载数组
data = np.load('test00000001.npy')
print(data) 输出: [1 2 3 4 5]
保存为`.npz`文件
import numpy as np
创建一个数组
list1 = [1, 2]
保存数组到文件
np.save('save_exam0.npy', list1)
从文件加载数组
npzfile0 = np.load('save_exam0.npy')
print(npzfile0) 输出: [1 2]
保存多个数组到`.npz`文件
import numpy as np
创建多个数组
arr1 = np.arange(100).reshape(10, 10)
arr2 = np.arange(100, 200).reshape(10, 10)
保存多个数组到文件
np.savez('save_exam0.npz', arr1=arr1, arr2=arr2)
从文件加载多个数组
npzfile0 = np.load('save_exam0.npz')
print(npzfile0['arr1']) 输出: array([[ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9],
print(npzfile0['arr2']) 输出: array([[100, 101, 102, 103, 104, 105, 106, 107, 108, 109],
以上是保存和加载数组的一些方法。您可以根据需要选择合适的方法