Python-matplotlib | 이중 y축 그래픽 그리기(범례 설정)

Python当打之年

풀어 주다： 2023-08-09 16:06:30

앞으로

2385명이 탐색했습니다.

오늘은 Python의 matplotlib 라이브러리를 사용하여 이중 y축 그래프를 그리는 방법과 범례 설정을 소개하겠습니다. 질문이나 제안이 있으면 편집자에게 비공개 메시지를 보낼 수 있습니다.

렌더링 미리보기:

Python-matplotlib | 이중 y축 그래픽 그리기(범례 설정)

샘플 데이터:

df = pd.read_csv(&#39;jobdata.csv&#39;)

로그인 후 복사

Python-matplotlib | 이중 y축 그래픽 그리기(범례 설정)

1. 이중 Y축 선 차트

1. 직급 꺾은선형 차트

colors = ["#51C1C8", "#536D84","#E96279"]
plt.figure(figsize=(16, 8))
ax1 = plt.subplot(111)
ax1.set_ylim(0,1200)
lin0 = ax1.plot(x_data, y_data1, marker=&#39;o&#39;, color=colors[0], label=&#39;岗位数量&#39;) 
for x, y in enumerate(y_data1):
    plt.text(x - 0.2, y+5, y)
ax1.set_ylabel(&#39;岗位数量&#39;,fontsize=12)
plt.legend()
plt.title("各城市Java岗位数量")
plt.show()

로그인 후 복사

Python-matplotlib | 이중 y축 그래픽 그리기(범례 설정)

2.ax1.twinx(를 통해 y축을 추가합니다. ):

# 增加y轴
ax2 = ax1.twinx()

ax2.set_ylim(0,60)
lin1 = ax2.plot(x_data, y_data2, linestyle=&#39;--&#39;, marker=&#39;o&#39;, c=colors[1], label=&#39;平均最低薪资&#39;) 
for x, y in enumerate(y_data2):
    plt.text(x - 0.1, y+1, y)
lin2 = ax2.plot(x_data, y_data3, linestyle=&#39;--&#39;, marker=&#39;o&#39;, c=colors[2], label=&#39;平均最高薪资&#39;)
for x, y in enumerate(y_data3):
    plt.text(x - 0.1, y+1, y)
ax2.set_ylabel(&#39;平均薪资(万/年)&#39;,fontsize=12)
plt.legend()
plt.title("各城市Java岗位数量和薪资水平状况")
plt.show()

로그인 후 복사

Python-matplotlib | 이중 y축 그래픽 그리기(범례 설정)

重点：细心的小伙伴可能发现了图没有问题，但是右上角的图例只显示了平均最低薪资和平均最高薪资，但是岗位数量的图例并没有显示。

3. 单独设置图例

ax1.legend(loc=&#39;best&#39;)
ax2.legend(loc=&#39;best&#39;)

로그인 후 복사

Python-matplotlib | 이중 y축 그래픽 그리기(범례 설정)

看着感觉没什么变化，实际上仔细看会发现平均最低薪资、平均最高薪资、岗位数量三个图例都显示出来了，只不过岗位数量图例被盖住了，我们可以移动一下位置看看：

ax1.legend(loc=2)
ax2.legend(loc=1)

로그인 후 복사

Python-matplotlib | 이중 y축 그래픽 그리기(범례 설정)

这样看就比较直观了，但是我就想把三个图例放一起不可以吗？

当然可以!

3. 设置组合图例

lines = lin0+lin1+lin2
labs = [label.get_label() for label in lines]
plt.legend(lines,labs)

로그인 후 복사

Python-matplotlib | 이중 y축 그래픽 그리기(범례 설정)

大功告成！

但是！如果是柱状图＋折线图的情况，效果还一样吗？

2、双y轴柱状图+折线图

1. 修改岗位数量为柱状图

plt.figure(figsize=(16, 8))
a1 = plt.subplot(111)
a1.set_ylim(0,1200)
bar = a1.bar(x_data, y_data1, color=colors[0], label=&#39;岗位数量&#39;) 
for x, y in enumerate(y_data1):
    plt.text(x - 0.2, y+5, y)
a1.set_ylabel(&#39;岗位数量&#39;,fontsize=12)

...

lines = bar+lin1+lin2
labs = [label.get_label() for label in lines]
plt.legend(lines,labs)

로그인 후 복사

直接报错了！ Python-matplotlib | 이중 y축 그래픽 그리기(범례 설정)

Python-matplotlib | 이중 y축 그래픽 그리기(범례 설정)

프롬프트 유형이 일치하지 않습니다. 이는 막대 및 선 유형에 문제가 있는 것입니다. 소스 코드를 확인해 보겠습니다.

matplotlib.axes.Axes.plot:

Python-matplotlib | 이중 y축 그래픽 그리기(범례 설정)

matplotlib.axes. Axes.bar:

Python-matplotlib | 이중 y축 그래픽 그리기(범례 설정)

ax.plot은 Line2D 유형의 목록을 반환하고 ax.bar는 patches 유형의 튜플을 반환합니다.

근본 원인을 찾은 후에는 Line2D와 패치를 조합하여 만들면 됩니다.

2. 设置Line2D和patches的组合图例

legend_handles = [ 
    Line2D([], [], linewidth=1, ls=&#39;--&#39;, lw=2, c=colors[2], label=&#39;平均最高薪资&#39;),
    Line2D([], [], linewidth=1, lw=2, c=colors[1], label=&#39;平均最低薪资&#39;),
    patches.Rectangle((0, 0), 1, 1, facecolor=colors[0],label=&#39;岗位数量&#39;)
]
plt.legend(handles=legend_handles, loc=&#39;best&#39;, fontsize=14)

로그인 후 복사

效果：