首頁  >  文章  >  後端開發  >  Python-matplotlib | 繪製雙y軸圖形(圖例設定)

Python-matplotlib | 繪製雙y軸圖形(圖例設定)

Python当打之年
Python当打之年轉載
2023-08-09 16:06:302322瀏覽


今天要介紹給大家使用Python的matplotlib函式庫繪製雙y軸圖及其圖例設定問題,希望對大家有幫助,如有疑問或建議可以私訊小編。
效果圖預覽:

Python-matplotlib | 繪製雙y軸圖形(圖例設定)

#
範例資料:
#
df = pd.read_csv('jobdata.csv')
Python-matplotlib | 繪製雙y軸圖形(圖例設定)

1.雙y軸折線圖

1 . 崗位數量折線圖

Python-matplotlib | 繪製雙y軸圖形(圖例設定)

Python-matplotlib | 繪製雙y軸圖形(圖例設定)

#####
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='o', color=colors[0], label='岗位数量') 
for x, y in enumerate(y_data1):
    plt.text(x - 0.2, y+5, y)
ax1.set_ylabel('岗位数量',fontsize=12)
plt.legend()
plt.title("各城市Java岗位数量")
plt.show()
##################2. 透過ax1.twinx() 增加一個y軸:#########
# 增加y轴
ax2 = ax1.twinx()

ax2.set_ylim(0,60)
lin1 = ax2.plot(x_data, y_data2, linestyle='--', marker='o', c=colors[1], label='平均最低薪资') 
for x, y in enumerate(y_data2):
    plt.text(x - 0.1, y+1, y)
lin2 = ax2.plot(x_data, y_data3, linestyle='--', marker='o', c=colors[2], label='平均最高薪资')
for x, y in enumerate(y_data3):
    plt.text(x - 0.1, y+1, y)
ax2.set_ylabel('平均薪资(万/年)',fontsize=12)
plt.legend()
plt.title("各城市Java岗位数量和薪资水平状况")
plt.show()
## ##########
重点:细心的小伙伴可能发现了图没有问题,但是右上角的图例只显示了平均最低薪资和平均最薪资,但是岗位数量的图例并没有显示。

3. 单独设置图例

ax1.legend(loc='best')
ax2.legend(loc='best')

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='岗位数量') 
for x, y in enumerate(y_data1):
    plt.text(x - 0.2, y+5, y)
a1.set_ylabel('岗位数量',fontsize=12)

...

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

直接报错了!Python-matplotlib | 繪製雙y軸圖形(圖例設定)Python-matplotlib | 繪製雙y軸圖形(圖例設定)Python-matplotlib | 繪製雙y軸圖形(圖例設定)

Python-matplotlib | 繪製雙y軸圖形(圖例設定)

提示類型不一致,顯然是bar和line的型別問題,我們查一下原始碼:

#matplotlib.axes.Axes.plot:

Python-matplotlib | 繪製雙y軸圖形(圖例設定)

#matplotlib.axes.Axes.bar:

Python-matplotlib | 繪製雙y軸圖形(圖例設定)##

ax.plot回傳的是一個Line2D類型的清單,ax.bar回傳的是一個patches 類型的元組。
找到根本原因後,我們做一個line2D和patches的組合就可以了。

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

legend_handles = [ 
    Line2D([], [], linewidth=1, ls='--', lw=2, c=colors[2], label='平均最高薪资'),
    Line2D([], [], linewidth=1, lw=2, c=colors[1], label='平均最低薪资'),
    patches.Rectangle((0, 0), 1, 1, facecolor=colors[0],label='岗位数量')
]
plt.legend(handles=legend_handles, loc='best', fontsize=14)
效果:
Python-matplotlib | 繪製雙y軸圖形(圖例設定)
其他参数大家可以自行尝试修改,对比前后效果,加深理解。

以上是Python-matplotlib | 繪製雙y軸圖形(圖例設定)的詳細內容。更多資訊請關注PHP中文網其他相關文章!

陳述:
本文轉載於:Python当打之年。如有侵權,請聯絡admin@php.cn刪除