python - 绘制 n 个图形并将它们保存在 n 个不同的文件中

Question

首先，我想明确一点，我不是 Python 专家，但仍在学习如何使用 pandas。我浏览了较旧的帖子，但找不到合适的答案。

我一直在尝试编写 92 份合约的数据分析代码。对于它们中的每一个，我想绘制一个特定的分析（每次获取相同数据帧的一些列）并将每个分析保存在不同的文件夹中（分析 1、分析 2、...）。

到目前为止，我面临着许多困难。因此，在关注要绘制的内容之前，我想了解如何编写每次将每个绘图保存在不同的 .png 文件中的代码。我尝试过的代码似乎没有保存任何内容，因为当我转到该文件夹​​时它是空的。

感谢waykiki 的帮助，我已经能够更新我的代码。现在我知道如何创建与我生成的分析一样多的文件夹。然而，我似乎不明白如何对每次分析的 92 个图的绘图进行编码。我的代码现在看起来像这样：

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

# Folder in which I want the analyses to be saved
URL5 = r"C:\Users\A\AppData\Local\Programs\Python\Python39"
# 1 graph per ID_Contrat (thus, 92 graphs)
groups = outer_merged_df.groupby("ID_Contrat") #where outer_merged_df is my dataframe
# How to name each plot.
List_ID_Contrat = outer_merged_df["ID_Contrat"].tolist()

def create_plot(file_name, x, y):
    # Create your plot. It is my understanding that here I should just give the x and the y I want to plot.
    fig = plt.figure()
    plt.plot(x, y, color = "red", kind = "line", legend = "true", linewidth = 2)
    plt.savefig(file_name)
    plt.show()

def main():
    # must be full-path. 
    parent_folder = URL5
    # move to parent directory
    os.chdir(parent_folder)
    # I want file_name to be different for each graph
    extension = ".png"
    # 5 = how many analyses I want to do
    for i in range(5):
        for name in List_ID_Contrat :
            file_name = "Analyse" + str[i+1] "{}" + extension.format(name) # I want file_name to be different for each graph and looking like "Analyse i Contrat XX"
        # Create a new folder
        folder_name = 'Analysis ' + str(i+1)
        os.mkdir(folder_name)
        full_file_name = folder_name + '/' + file_name
        x = np.linspace(1,100,100)
        y = np.random.random(100)
        create_plot(full_file_name, x, y)
        print("plot "+ savefile +" finished".format(name))
        
if __name__ == "__main__":
    main()

然而，当我运行我的代码时，它不再绘制 92 个图，也不想再创建文件夹（尽管它确实使用了 Waykiki 的方法）。在第一轮中 for 循环被破坏（我只得到文件夹“分析 1”）我收到错误消息：

AttributeError: 'Line2D' object has no property 'kind'

您能否向我解释一下如何保存图表？我迷路了..

谢谢

标签： pythonloopsmatplotlibsavefig

Answer 1

我认为您的方法是正确的，因为您将问题分为两个步骤：

1.) 完成技术细节（创建、组织和浏览文件夹和数据）。

2.) 实际创建/绘制地块。

这是一个简单的原型脚本。该脚本在主目录'/home/user/my_analysis/'中创建 N 个子文件夹。所有子文件夹都命名为“AnalysisX”，其中 X 是文件夹的编号。

每个文件夹都包含不同的情节。

注意：我的文件夹路径适用于 linux 机器，所以请记住，“/home/user/some_folder/”在 Windows 中不是有效路径！（我看到你已经把那部分做对了，但它可能对其他用户有用）。

import os
import numpy as np
import matplotlib.pyplot as plt


def create_plot(file_name, x, y):
    # Create your plot
    fig = plt.figure()
    plt.plot(x, y, color='red', linewidth=2)
    plt.savefig(file_name)
    plt.show()


def main():
    # must be full-path
    parent_folder = '/home/user/my_analysis/'

    # move to parent directory
    os.chdir(parent_folder)

    file_name = 'plot'
    extension = '.png'
    for i in range(5):
        # Create a new folder
        folder_name = 'Analysis' + str(i+1)
        os.mkdir(folder_name)

        full_file_name = folder_name + '/' + file_name + extension
        x = np.linspace(1, 100, 100)
        y = np.random.random(100)
        create_plot(full_file_name, x, y)


if __name__ == '__main__':
    main()

为清楚起见，这就是文件夹结构的样子。我只审查了我的真实用户名：