如何为分组的熊猫数据框添加标题？

Question

我正在尝试更改由分组 pandas 数据框制作的直方图的标题、y 轴标题和 x 轴编号。这是我所说的一个例子：

%matplotlib inline
import matplotlib.pyplot as plt
import pandas as pd
import seaborn as sns
from random import randint
plt.style.use('seaborn-whitegrid')

a = []
for i in range(200):
    a.append(randint(0, 1))
b = []
for i in range(200):
    b.append(randint(1, 3))

d = {'A': a, 'B': b}
df = pd.DataFrame(data=d)

df_group = df.groupby(['A'], as_index = True)

df_group.hist(bins=5)

这是我在运行代码时看到的： Original Plot

我想做的是在python中编辑它看起来像这样（我在paint中做了这个）： Edited Plot

任何帮助将不胜感激。

Answer 1

这应该得到你想要的结果。

%matplotlib inline
import matplotlib.pyplot as plt
import pandas as pd
import seaborn as sns
from random import randint
plt.style.use('seaborn-whitegrid')

a = []
for i in range(200):
    a.append(randint(0, 1))
b = []
for i in range(200):
    b.append(randint(1, 3))

d = {'A': a, 'B': b}
df = pd.DataFrame(data=d)

df_group = df.groupby(['A'], as_index = True)

ax = df_group.hist(bins=5)
for i in range(len(ax)):
    ax[i][0][0].set_title(f"A-{i}")

我只是遍历子图并手动重命名它们。