わかさぎのブログ

プログラミング、Atcoderの勉強とか

大量のグラフを作成し保存

   def picture_make(file_name):

    #画像保存用のディレクトリを作成しその中にグラフの画像を保存する
        picture_title=file_name[-8:-4]
        picture_box=os.path.join(path_loc,"picture")

        if os.path.exists(picture_box)==False:
            os.mkdir(picture_box)
        else:
            pass

        plt.savefig(os.path.join(picture_box, picture_title))

 

解説

ファイル名に対してスライスで必要な部分だけ取り出す。

os.path.joinで保存先とタイトルを連結する。

 

ファイル名は、ファイル名一覧のリストを使ってfor文で回す。

for file in files:

    picture_make(file)

みたいにする。