Python中将列表保存并加载

将实验过程中列表类型的数据保存，用于后期实验复现。

import os 
import numpy as np

train_acc = [0.5,0.7,0.75,0.8,0.85,0.88,0.87,0.89,0.85,0.9,0.86,0.9]

# 创建文件夹，存放输出结果
if not os.path.exists("result_variable"):
    os.mkdir("result_variable")
# 保存数据    
np.savez('result_variable/train_acc',train_acc)

# 加载数据
train_acc = np.load('result_variable/train_acc.npz')
print(train_acc.files) # 查看各个数组名称 
'''
['arr_0']
'''

train_acc = train_acc['arr_0']#object 只读
print(len(train_acc))
'''
12
'''

''' 画出曲线'''     
import matplotlib.pyplot as plt

plt.figure(1)
# plt.plot(np.arange(0,len(train_acc),1), train_acc, label='train acc')
plt.plot(np.arange(len(train_acc)), train_acc, label='train acc')
plt.title('train acc') 
plt.xlabel("epoch")
plt.ylabel("acc")
plt.legend()
plt.savefig('result_variable/temp.png',dpi=600) # 保存图片
plt.show()

注：横坐标的设置，请看这篇博客：设置绘制图片的横坐标，这里不再赘述。