对pandas的dataframe绘图并保存的实现方法

ax = df.plot() 
fig = ax.get_figure()
fig.savefig('fig.png')

label_dis = df.label.value_counts()
ax = label_dis.plot(title='label distribution', kind='bar', figsize=(18, 12))
fig = ax.get_figure()
fig.savefig('label_distribution.png')