Python 如何根据类别设置不同绘图的图例?

Python 如何根据类别设置不同绘图的图例?,python,dataframe,matplotlib,legend,Python,Dataframe,Matplotlib,Legend,我有一个很大的数据集,超过7000行和300列。我正在处理诊断和住院时间。我想为每个诊断做一个绘图(因此在xlabel中,条数将与患者的条数相同,而在ylabel中,是每个患者进行该诊断的天数) 但是,我希望每个图都有一个不同的标题,以便了解该图所指的每个诊断。我的意思是,例如,对于心脏病发作,我希望标题是“心脏病发作”,但对于不同的诊断,我希望另一个标题。我希望它能自动完成。可能吗?我在下面显示了我的代码,但我知道它不正确 此外,我希望xlabel和ylabel出现在每个绘图中,知道它没有发生

我有一个很大的数据集,超过7000行和300列。我正在处理诊断和住院时间。我想为每个诊断做一个绘图(因此在
xlabel
中,条数将与患者的条数相同,而在
ylabel
中,是每个患者进行该诊断的天数)

但是,我希望每个图都有一个不同的标题,以便了解该图所指的每个诊断。我的意思是,例如,对于心脏病发作,我希望标题是“心脏病发作”,但对于不同的诊断,我希望另一个标题。我希望它能自动完成。可能吗?我在下面显示了我的代码,但我知道它不正确

此外,我希望
xlabel
ylabel
出现在每个绘图中,知道它没有发生什么。它只是出现在最后一个

有人能帮我吗

df_n = df[['Diagnose','Length of Stay']]
df_n.groupby('Diagnose').plot(kind = 'bar')
plt.xlabel('Patients')
plt.ylabel('Length of Stay (Days)')
plt.title('Length of Stay for Diagnose') # I want it to be the name of the diagnose
plt.show()

我相信您的标签没有显示出来,因为您没有为您将要制作的每个绘图明确设置它们。 当我做分组时,我喜欢编字典。每个诊断将是一个键,值将是数据帧

a = dict(tuple(df_n.groupby('Diagnose')))
for key, value in a.items():
     value.plot(kind = 'bar')
     plt.title('{}'.format(key))
在for循环下,您可以一次性设置所有格式,它将为everygraph执行此操作。您可以输入
plt.savefig({0}.eps'.format(k)
语句,这将保存所有绘图