我有这个小脚本来制作一个图:
plt.plot(loss_test3)
plt.plot(loss)
plt.legend(['loss_test3','loss_test4'], loc='upper left')
t = np.linspace(loss.min(), loss_test3.max(), 5)
plt.yticks(t, t)
plt.xlim(0, 200)
plt.title('test 4 loss')
plt.ylabel('loss')
plt.xlabel('epoch')
plt.savefig('test_4_loss.png', dpi=100)
plt.show()
我得到的输出是这样的:
如何舍入Y轴上的值?
发布于 2019-06-13 07:39:57
我是这样解决这个问题的:
from matplotlib.ticker import StrMethodFormatter
plt.plot(loss_test3)
plt.plot(loss)
plt.legend(['loss_test3','loss_test4'], loc='upper left')
#t = [loss.min(), loss.max()]
t = np.linspace(loss.min(), loss_test3.max(), 5)
plt.yticks(t, t)
plt.gca().yaxis.set_major_formatter(StrMethodFormatter('{x:,.3f}'))
plt.xlim(0, 200)
plt.title('test 4 loss')
plt.ylabel('loss')
plt.xlabel('epoch')
plt.savefig('test_4_loss.png', dpi=100)
plt.show()
https://stackoverflow.com/questions/56571532
复制相似问题