有啥方法可以在for循环中保存多个图而不用python覆盖？

Posted 2023-02-23

【中文标题】有啥方法可以在for循环中保存多个图而不用python覆盖？

【英文标题】：any way to save multiple plots in for loop without overwriting in python?有什么方法可以在for循环中保存多个图而不用python覆盖？

【发布时间】：2020-06-03 19:53:37

【问题描述】：

我有一个汇总数据框列表，其中包含不同国家/地区的贸易统计数据。我可以通过遍历这个数据框列表来制作相应的图。现在我打算将它们保存在本地而不覆盖，我的意思是用各自的标题名称保存每个图。为此，我尝试了以下方法：

我有以下名称的数据框列表：

[dfList[i].name for i in range(len(dfList))]

['AR  fresh-Beef-E',
 'AUC  fresh-Beef-E',
 'BR  fresh-Beef-E',
 'CA  fresh-Beef-E',
 'CL  fresh-Beef-E',
 'CN  fresh-Beef-E',
 'E28  fresh-Beef-E',
 'EG  fresh-Beef-E',
 'IN  fresh-Beef-E',
 'JP  fresh-Beef-E',
 'KR  fresh-Beef-E',
 'MX  fresh-Beef-E',
 'NZ  fresh-Beef-E',
 'PY  fresh-Beef-E',
 'US  fresh-Beef-E',
 'UY  fresh-Beef-E',
 'ZA  fresh-Beef-E']

当前尝试：

打算在本地保存图形，并将其绘图标题作为文件名：

import os
import pandas as pd
import matplotlib.pyplot as plt

outpath = r'C:/Users/plots'

if os.path.exists(outpath):
    shutil.rmtree(outpath)
_ = os.mkdir(outpath)

for i in range(len(dfList)) :
    plt.figure()
    my_plotter(dfList[i],title=dfList[i].name)
    plt.savefig(path.join(outpath,"dataname_0.png".format(i)))
    plt.close()

新更新

这是我的绘图功能的样子：

def my_plotter(df, plot_type='something', ylab_nm='something', title=None):
    fig, ax1 = plt.subplots(figsize=figsize)
    if plot_type=='something':
        _ = df.plot(kind='line', ax=ax1, marker='o', ls='-', linewidth=2, color=colors)
    else:
        df.loc[:, 'Total'] = df.sum(axis=1)
        _ = df.div(df.Total, axis=0).iloc[:, :-1].plot(kind='line', ax=ax1, marker='o', ls='--', linewidth=4, color=colors)
        df.drop('Total', axis=1, inplace=True)
        ax1.yaxis.set_major_formatter(mtick.PercentFormatter(xmax=1, decimals=0))
        ax1.xaxis.set_major_locator(mdates.MonthLocator(bymonthday=1, interval=3))

    ax1.set(title=title)
    plt.title(title)
    ax1.xaxis.label.set_visible(False)
    plt.style.use('ggplot')
    plt.xticks(rotation=90)
    plt.show()

但上述尝试并未将绘图保存到本地目录。我查看了SO 并尝试了可能的建议，但我仍然没有得到所有应该通过获取其绘图标题来命名并保存到本地文件目录的绘图。在我的尝试中，没有一个图被保存到本地目录。

谁能指出我如何做到这一点？任何想法？ 目标

我打算通过将其标题作为文件名来保存每个图，并将其保存到本地目录。有什么想法吗？谢谢

【参考方案1】：

这不是一个完整的 MWE，但我认为下面的修复应该可以帮助您。

import matplotlib.pyplot as plt
import pandas as pd
import os

# Only need to set the style once (not in loop)
plt.style.use('ggplot')

outpath = r'C:/Users/plots'

def my_plotter(df, plot_type='something', ylab_nm='something', title=None):
    fig, ax1 = plt.subplots(1, 1)
    if plot_type=='something':
        # Plots on ax1, so don't save output as another ax instance
        df.plot(kind='line', ax=ax1, marker='o', ls='-', linewidth=2)
    else:
        df.loc[:, 'Total'] = df.sum(axis=1)
        # Plots on ax1, so don't save output as another ax instance
        df.div(df.Total, axis=0).iloc[:, :-1].plot(kind='line', ax=ax1, marker='o', ls='--', linewidth=4, color=colors)
        df.drop('Total', axis=1, inplace=True)
        ax1.yaxis.set_major_formatter(mtick.PercentFormatter(xmax=1, decimals=0))
        ax1.xaxis.set_major_locator(mdates.MonthLocator(bymonthday=1, interval=3))

    ax1.set(title=title)
    ax1.set_title(title, size=24, verticalalignment='bottom') 
    ax1.xaxis.set_major_formatter(mdates.DateFormatter('%b %Y'))
    ax1.xaxis.label.set_visible(False)

    plt.xticks(rotation=90)
    plt.show()
    return fig, ax1

if os.path.exists(outpath):
    shutil.rmtree(outpath)
_ = os.mkdir(outpath)

for i in range(len(dfList)):
    fig, ax1 = my_plotter(dfList[i], title=dfList[i].name)
    fig.savefig(path.join(outpath,"dataname_.png".format(dfList[i].name)))
    plt.close()

【讨论】：

我在尝试此操作时遇到类型错误：TypeError: join() takes exactly one argument (2 given)。此尝试也不会通过将绘图标题名称作为文件名来保存图形。有更好的主意吗？

在脚本顶部尝试import os.path as path

@Ralph 现在保存了但是变成了文件大小2kb 只表示实际生成的数字没有正确保存。这是为什么？有更好的主意吗？

它们只是空的数字和轴吗？ my_plotter 是做什么的？

是的，是空图。 my_plotter为时间序列数据制作折线图。