python - 如何跨子图添加层次轴以标记组？

Question

我有一组可以分组的不同时间序列。例如。下图显示了系列 A、B、C 和 D。但是，A 和 B 在组 G1 中，而 C 和 D 在组 G2 中。

我想在图中通过在左侧添加另一个轴来反射(reflect)这一点，该轴穿过涡轮机组并相应地标记这些轴。

到目前为止，我已经尝试了一些东西，但显然其中一个并不那么容易。

有人知道我该怎么做吗？

PS: 因为我在一个已经有列的数据框上使用 Pandas 的plot(subplots=True)

      |  G1   |  G2  |
      |-------|------|
index | A  B  | C  D |
------|-------|------|

pandas 可能已经为我做到了。这就是我使用 pandas 的原因 标签。

Answer 1

您可以在图中创建额外的轴，它跨越每两个图，但只有一个左侧的 y 轴，没有刻度和其他装饰。只设置了一个 ylabel。这将使整个事情看起来很协调。

好处是您可以使用现有的 pandas plot。缺点是多了15行代码。

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.gridspec as gridspec


df = pd.DataFrame(np.random.rand(26,4), columns=list("ABCD"))
axes = df.plot(subplots=True)
fig = axes[0].figure


gs = gridspec.GridSpec(4,2)
gs.update(left=0.1, right=0.48, wspace=0.05)
fig.subplots_adjust(left=.2)

for i, ax in enumerate(axes):
    ax.set_subplotspec(gs[i,1])

aux1 = fig.add_subplot(gs[:2,0])
aux2 = fig.add_subplot(gs[2:,0])

aux1.set_ylabel("G1")
aux2.set_ylabel("G2")

for ax in [aux1, aux2]:
    ax.tick_params(size=0)
    ax.set_xticklabels([])
    ax.set_yticklabels([])
    ax.set_facecolor("none")
    for pos in ["right", "top", "bottom"]:
        ax.spines[pos].set_visible(False)
    ax.spines["left"].set_linewidth(3)
    ax.spines["left"].set_color("crimson")


plt.show()