python matplotlib "merge"用于多索引的x轴标签

Question

我想“合并”多索引的第一级，即有一个居中的“USA”和“EU”而不是元组。

最小示例

df = pd.DataFrame(data={"region":["USA","USA","EU","EU"], "sctr":["HH","NFC","HH","NFC"], "values":[1,2,3,4]})
df = df.set_index(["region","sctr"])

fig, ax = plt.subplots(1,figsize=(8,6))
df.plot(kind="bar")
plt.xticks(rotation = 0)
plt.show()

输出

所需输出 使用 Excel，默认情况下(几乎)可以获得所需的输出:

理想的解决方案是简单、简短，并且在添加更多区域/扇区时不需要任何手动调整。

Answer 1

还有其他方法可以将x轴刻度分成两行:第一行从多索引中取出并制成列表；第二行由文本函数从多索引列表中放置。隐藏现有标签由绘图参数处理。

import matplotlib.pyplot as plt
import pandas as pd

df = pd.DataFrame(data={"region":["USA","USA","EU","EU"], "sctr":["HH","NFC","HH","NFC"], "values":[1,2,3,4]})
df = df.set_index(["region","sctr"])

fig, ax = plt.subplots(1,figsize=(8,6))
df.plot(kind='bar', xlabel='', ax=ax)
ax.set_xticklabels(df.index.get_level_values(level=1).tolist(), rotation=0)

 for container in ax.containers:
    for i,child in enumerate(container.get_children()):
        if i == 0:
            ax.text(child.xy[0]+child.get_width(), -0.08, df.index.get_level_values(level=0)[0], ha='center', transform=ax.transAxes)
        elif i == 2:
            ax.text(child.xy[0]-(child.get_width()*2), -0.08, df.index.get_level_values(level=0)[2], ha='center', transform=ax.transAxes)

plt.show()