使用for循环和pandas将fig插入excel工作簿

时间:2016-09-16 15:07:51

标签: python pandas for-loop xlsxwriter

我创建了一个for循环,我将数据帧写入excel文件,然后绘制datafame并将图像保存为png文件。在循环中,我想将每个保存的图像添加到每个Excel工作表中。我知道如何使用openplyx在循环外执行此操作,但这将是太多不必要的代码。

如何将无花果插入工作簿中的每个工作表?

这是我的代码:

writer = pd.ExcelWriter('america.xlsx')
countrylist = countrylist.set_index(['COUNTRY_NAME'])
for i in countrylist.index:
    var = i
    test = likes[likes['COUNTRY_NAME'].map(str.strip)== str((var))]
    test.rename(columns={'YEAR': 'Year'}, inplace=True)
    test.to_excel(writer, var, index=False, columns = ['Year', 'Area', 'Region'])
    ax = test.plot.bar('Year', ['Area', 'Region'], align = 'center', alpha=0.5, title=var)
    fig = ax.get_figure()
    folder = 'ra/'
    fig.savefig(folder + var + '.png')

0 个答案:

没有答案