用python和matplotlib画双坐标系曲线

最新推荐文章于 2022-10-11 19:30:30 发布

原创最新推荐文章于 2022-10-11 19:30:30 发布 · 1.9k 阅读

7 ·

CC 4.0 BY-SA版权

文章标签：

#matplotlib #python

写论文技巧专栏收录该内容

2 篇文章

订阅专栏

本文介绍了一种使用Matplotlib和NumPy在Python中绘制带有双Y轴的曲线图的方法，并展示了如何通过设置不同的颜色和标签来区分两条曲线，适用于展示对比实验结果。

有时候写论文的时候不免会遇到画曲线图，而且还有可能需要画双坐标系（双Y轴）的曲线图，在百度中找了许久，先mark一下，以下代码就是我论文中用到的，以后应该还会用到，先保存在这里。结果图如下：

import matplotlib.pyplot as plt
import numpy as np
import scipy.io as sio

x = np.arange(0., 1000, 2, dtype=np.int32)
xx = np.arange(0., 2000, 4, dtype=np.int32)
data = sio.loadmat('C://Users//gmt//Desktop//bn.mat')
y1 = data["loss"][:1000, 0]
y2 = data["testacc"][:1000, 0]
data = sio.loadmat('C://Users//gmt//Desktop//wobn.mat')
y11 = data["loss"][:1000, 0]
y22 = data["testacc"][:1000, 0]

fig = plt.figure()

ax1 = fig.add_subplot(111)
ax1.plot(xx, y1[x], color='red', label='Loss with BN')
ax1.plot(xx, y11[x], color='green', label='Loss without BN')
ax1.set_ylabel('Cross Entropy Loss')
ax1.set_xlabel('Iterations')
ax1.set_ylim(0, 1500)
ax1.set_xlim(0, 1000)
plt.legend(loc='lower left')


ax2 = ax1.twinx()
ax2.plot(xx, y2[x], color='blue', label='Test Accuracy with BN')
ax2.plot(xx, y22[x], color='skyblue', label='Test Accuracy without BN')
ax2.set_ylabel('Testset Accuracy')
ax2.set_xlabel('Iterations')
ax2.set_ylim(0, 1)
ax2.set_xlim(0, 1000)
plt.legend(loc='upper right')
plt.show()