Matplotlib:如何在两个独立的轴上显示条形图和直线的图例?

mi7gmzs6  于 2023-10-24  发布在  其他
关注(0)|答案(2)|浏览(125)

我在左轴上绘制一个条形图,在右轴上绘制一个折线图。如何在同一个图例框中显示两个图的图例?
使用下面的代码,我得到了两个单独的图例框;此外,我需要手动指定第二个图例的位置,这需要大量的试错,否则它会与第一个图例重叠。

import pandas as pd
import matplotlib.pyplot as plt
import matplotlib

df=pd.DataFrame()
df["bar"]=[40,35,50,45]
df["line"]=[.5,.3,.2,.6]

fig,ax=plt.subplots(2)

l1 = ax[0].bar(df.index, df["bar"], label='my bar chart (left axis)', color='royalblue', width = 0.55)
ax0b=ax[0].twinx()
l2 = ax0b.plot(df.index,  df['line'], label='my line (right axis)',color='tomato',marker='.', ls='dashed')

ax[0].legend(loc='upper left', fancybox=True, title='My title')
ax0b.legend(bbox_to_anchor=(0.155,0.8), loc=1)    
plt.show()

如果我有两条线在两个不同的轴上,而不是条形图和折线图,我会这样做:

myl=l1+l2
labs=[l.get_label() for l in myl]
ax[0].legend(myl, labs, loc='upper left')

但这对我来说不起作用。我得到:

myl=l1+l2
TypeError: can only concatenate tuple (not "list") to tuple

我想这一定是因为bar()和plot()返回了两个不同的对象,它们不能连接在一起。

vkc1a9a2

vkc1a9a21#

python错误的好处是,它们通常可以从字面上理解并直接告诉你问题。
“TypeError”告诉你你有一个类型问题。你可以打印类型:

print(type(l1)) # <class 'matplotlib.container.BarContainer'>
print(type(l2)) # <type 'list'>

现在,“can only concatenate tuple(not“list”)to tuple”告诉您不能将barcontainer添加到列表中。
简单的解决方案:添加两个列表:

myl=[l1]+l2
swvgeqrz

swvgeqrz2#

这对我很有效:ax -轴/子图为bar,ax 1-为line
handles1, labels1 = ax.get_legend_handles_labels()
handles2, labels2 = ax1.get_legend_handles_labels()
ax.legend((*handles1, *handles2), (labels1, labels2))

相关问题