在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],
以上是保存和加载数组的一些方法。您可以根据需要选择合适的方法

