Pythonでディレクトリを新規作成してmatplotlibのグラフをファイル保存する

ディレクトリを新規作成してグラフをファイル保存

ディレクトリを新規作成するには、osモジュールをimportします。

そして、makedirs()でディレクトリを新規作成します。

exist_ok=Trueとしておくことで、既に指定した名前のディレクトリがある場合にも、エラーとなりません。
スクリプトを複数回実行する場合には、指定しておくと良いかと思います。

import os

dirname = "dir001/"
os.makedirs(dirname, exist_ok=True)

グラフをファイル保存する場合には、matplotlib.pyplotのsavefig()を使えばOKです。

filename = dirname + "img.png"
plt.savefig(filename)

サンプルコード

下記がサンプルコードになります。

 $ cat sample.py 
#!/usr/bin/env python3
# coding: UTF-8

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

dirname = "dir001/"
os.makedirs(dirname, exist_ok=True)

data = pd.read_csv('in.txt',
				names=('Date', 'Value1'),
				index_col='Date',
				parse_dates=True)

print("dataの中身を出力")
print (data)

plt.plot(data.Value1)

filename = dirname + "img.png"
plt.savefig(filename)

plt.show()

実行すると、./dir001/img.pngができあがります。