for文で行列（.csvファイル）を読み込んで表示させた画像を連続保存したい

pythonでフォルダから.csvファイルの行列を読み込んで画像表示させているのですが、表示させた画像を連続保存する方法ってありますか？
plt.savefig() を使用させてみたのですが、計算させた最後のファイルだけが保存される状況です。


import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import os, glob

dir = '~/2D_Dataset/train/One/*.csv'

df_list=[]
book_dict = {}
for i in glob.glob(dir):
    file_name = os.path.basename(i)
    book_dict[file_name] = print(i)
    data=pd.read_csv(i, header=None, skiprows=0)
    df_list.append(data)
for j in df_list:
    print(j)
    plt.figure()
    plt.imshow(j)
    plt.savefig('j.png')

結果は以下のように出力されます。

  0     1     2     3     4     5

0 0.31 0.12 0.40 0.41 1.00 0.24
1 0.00 0.00 0.31 0.40 0.00 0.00
2 0.88 0.34 0.68 0.39 0.48 1.00
3 0.38 0.32 0.13 0.46 0.28 0.21
4 0.70 0.73 0.60 0.35 0.41 0.76
5 0.70 0.71 0.86 0.35 0.59 0.68
0 1 2 3 4 5
0 0.70 0.71 0.86 0.35 0.59 0.68
1 0.37 0.37 0.11 0.02 1.00 0.18
2 0.31 0.15 0.30 0.35 0.24 0.38
3 0.38 0.32 0.13 0.46 0.28 0.21
4 0.70 0.71 0.86 0.35 0.59 0.68
5 0.38 0.32 0.13 0.46 0.28 0.21
0 1 2 3 4 5
0 0.70 0.71 0.86 0.35 0.59 0.68
1 0.72 0.27 0.39 0.34 0.28 0.62
2 0.88 0.34 0.68 0.39 0.48 1.00
3 1.00 0.46 0.86 0.40 0.59 0.88
4 1.00 0.46 0.86 0.40 0.59 0.88
5 0.37 0.37 0.11 0.02 1.00 0.18
![]

このように出力される画像をファイルとして個別に保存したいのですが、matplotlibで出来ますでしょうか。

行動規範の内容に同意します

回答2件

ベストアンサー

質問に記述したコードを下のように修正して、CSVのファイル名と同じ名前で画像を保存するようにするのはどうでしょうか

Python
1dir = '~/2D_Dataset/train/One/*.csv'
2for i in glob.glob(dir):
3    file_name = os.path.basename(i)
4    # ファイル名から拡張子を分離
5    basename,_ = os.path.splitext(file_name)
6    data=pd.read_csv(i, header=None, skiprows=0)
7    plt.figure()
8    plt.imshow(data)
9    # タイトルを描画(追加)
10    plt.title(basename)
11    # CSVと同じ名前で画像を保存
12    plt.savefig(f"{basename}.png")
13    # グラフを削除（追加）
14    plt.clf()