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

Question

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

我一直在尝试对 92 份契约(Contract)的数据分析进行编码。对于它们中的每一个，我想绘制一个特定的分析(每次都采用相同数据帧的一些列)并将每个分析保存在不同的文件夹中(分析 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循环在第一轮中被打破(我只得到文件夹“Analysis 1”)我收到错误消息:

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

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

谢谢

Answer 1

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

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

2.) 实际创建/绘制绘图。

这是一个简单的原型(prototype)脚本。此脚本在主目录 '/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()

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