如何为分组的pandas数据框添加标题?

时间:2018-01-08 01:11:11

标签: python pandas group-by histogram

我正在尝试更改由分组的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

任何帮助都将不胜感激。

1 个答案:

答案 0 :(得分:0)

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

%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}")

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