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

2024-04-25 06:21:52 发布

您现在位置:Python中文网/ 问答频道 /正文

我创建了一个for循环,在这个循环中,我将一个dataframe写入一个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')

Tags: 文件nametest图像forindexpngvar