双轴图中添加图例：解决图例缺失难题

python

2024-03-20 07:48:23

在双轴图中优雅地添加图例

问题：图例中的缺失

在绘制双轴图时，你可能会遇到一个令人头疼的问题：第二个 y 轴的标签无法显示在图例中。这是因为 twinx() 方法创建了一个新的轴，它与主轴共享 x 轴，但拥有独立的 y 轴。

解决方案：双轴图例的魔法

为了解决这个难题，你需要一个咒语：

绘制所有数据： 首先，绘制你想要显示的所有数据，并确保为每个序列指定标签。
创建双轴： 用 twinx() 方法召唤第二个 y 轴。它将与主轴共享 x 轴，但拥有自己的 y 轴刻度。
添加图例： 现在，施展 legend() 魔法。将 handles 参数设置为包含所有要显示在图例中的线条句柄的列表。
添加第二个 y 轴的句柄： 为了让第二个 y 轴的标签出现在图例中，使用 twinx() 方法返回的第二个轴对象。

代码咒语

以下代码将释放双轴图例的魔力：

import numpy as np
import matplotlib.pyplot as plt

# 绘制数据
time = np.linspace(0, 24, 100)
Swdown = np.sin(time)
Rn = np.cos(time)
temp = np.random.rand(100)

# 创建双轴图
fig, ax = plt.subplots()
ax.plot(time, Swdown, label="Swdown")
ax.plot(time, Rn, label="Rn")
ax2 = ax.twinx()
ax2.plot(time, temp, color='r', label="Temp")

# 添加图例
handles1, labels1 = ax.get_legend_handles_labels()
handles2, labels2 = ax2.get_legend_handles_labels()
ax.legend(handles1 + handles2, labels1 + labels2, loc=0)

# 完善图表
ax.set_xlabel("Time (h)")
ax.set_ylabel("Radiation (MJ/m2/d)")
ax2.set_ylabel("Temperature (°C)")
plt.show()