如何在Matplotlib的水平条形图上绘制次坐标轴纵向折线?
实现水平条形图+纵向次坐标轴折线
当然可以实现,核心是要手动创建共享y轴的次坐标轴(而非pandas默认的共享x轴),具体修改如下:
修改后的代码
import matplotlib.pyplot as plt import pandas as pd df = pd.DataFrame( { "A": [2, 4, 8], "B": [3, 5, 7], "C": [300, 100, 200], }, index=["a", "b", "c"], ) # 创建主坐标轴并绘制水平条形图 fig, ax0 = plt.subplots(figsize=(10.24, 5.12)) df.plot(y=df.columns[:2], kind="barh", ax=ax0) # 创建共享y轴的次坐标轴(twiny() 共享y轴,独立x轴) ax1 = ax0.twiny() # 在次坐标轴上绘制纵向折线:x为C的数值,y为类别索引对应位置 ax1.plot(df["C"], df.index, color="g", marker=".") # 设置次坐标轴标签 ax1.set_xlabel("C") plt.tight_layout() plt.show()
关键说明
直接替换bar为barh后折线仍横向的原因:
pandas的plot(secondary_y=True)默认会创建twinx()(共享x轴、独立y轴)的次坐标轴,这适配垂直条形图的逻辑,但对水平条形图来说,我们需要的是共享y轴、独立x轴的次坐标轴(用twiny()创建),这样折线的x轴对应C的数值,y轴和主条形图的类别对齐,最终呈现出纵向的折线效果,等同于原垂直图表旋转90度。
内容的提问来源于stack exchange,提问作者chaosink
相关产品推荐
相关产品推荐

