出力したグラフを指定のディレクトリに保存する方法

Question

前提・実現したいこと

pythonで、任意のディレクトリにあるファイルを使ってループ処理でグラフを作成するプログラムを書いています。
ループ処理の度に出力したグラフを指定のディレクトリに、保存したいと思っています。

グラフを保存する機能を実装中に以下のエラーメッセージが発生しました。

発生している問題・エラーメッセージ

OSError: [Errno 22] Invalid argument: 'C:/results/plotC:/results¥¥data_100.png'

該当のソースコード

files=glob.glob(r'C:/*.csv')
save_dir = 'C:/results'


for file in files:
   filename, file_extension = os.path.splitext(file)

   for x in range(a,b):
       df = pd.read_csv(file, header=None, usecols=[x])

       plt.plot(df,marker="+", color='b', ls="", markersize=10)  
       plt.savefig(os.path.join(save_dir) + 'plot'+str(filename)+str(x)+'.png') 
       plt.close()

試したこと

/を¥に変更など。

補足情報（FW/ツールのバージョンなど）

jupyter、python、windows10

Answer 1

原因

os.path.splitext はファイルパスを (拡張子より手前の部分, 拡張子) に分ける関数なので、filename はファイル名ではなく、ファイルパスになっています。

import os

filename, file_extension = os.path.splitext(r"C:\hoge\fuga.txt")
print(filename, file_extension)  # C:\hoge\fuga .txt

これを

os.path.join(save_dir) + 'plot'+str(filename)+str(x)+'.png'

で結合しているため、'C:/results/plotC:/results¥¥data_100.png' となります。
(各変数を print してどうなっているか確認してみてください。)

対応策

拡張子を除いたファイル名を得るには、さらに os.path.basename(filename) とする必要があります。

Comments

[cacca]

早々にコメントいただき、ありがとうございます。os.pathについてよく理解していませんでした。

plt.savefig(save_dir, 'plot'+'os.path.basename(filename)'+str(x)+'.png')

こう書き換えてみたのですが、
TypeError: print_figure() got multiple values for argument 'dpi'
というエラーになってしまいました。
初心者で申し訳ございませんが、ご助言いただけましたら幸いです。

[tiitoi]

plt.savefig(os.path.join(save_dir, 'plot'+'os.path.basename(filename)'+str(x)+'.png'))
ではないでしょうか

[cacca]

ありがとうございます。お陰様で解決できました。しばらく悩んでいたので、大変助かりました。また機会がありましたら、ご助言お願い致します。