for in構文でループを行うたびにcsvファイル名を変え，指定のフォルダに保存したい

Question

前提・実現したいこと

for in 構文でフォルダ内のcsvファイルをループ処理しているのですが，
ループを繰り返すたびに，ファイルが上書きされてしまいファイルが
一つしか得られません。

if構文で条件を満たしたファイルを，指定したフォルダに
elseで条件を満たさなかったファイルを指定したフォルダにそれぞれ保存がしたいです。

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

エラーメッセージ

該当のソースコード

if len(card_and_measure) >=3:
print("成功")
data.to_csv('TA2.csv', index=False, encoding='SHIFT_JIS')
else:
print("不可")

import pandas as pd
import glob
import os

for file in glob.glob('TA*.csv'): #file内のcsvファイルを検索し全て読み混む
  
    basename = os.path.basename(file)
    
    name = os.path.splitext(basename)[0]
    
    data = pd.read_csv(file,
       names=["2θ", "Intensity"],
       encoding="SHIFT_JIS")
      
       
    mycsv = pd.read_csv('TA.csv', names=["2θ", "Intensity"], encoding="SHIFT_JIS")#ヘッダーがないため追加
    
    card = mycsv.sort_values(by="Intensity", ascending=False)#Intensityの行をascending=Falseによって降順に並び替え
    measure = data.sort_values(by="Intensity", ascending=False).head(50)#降順の上から100個のみを取り出し
    
    measure_x = measure.iloc[:,0]
    card_x = card.iloc[:,0]
    card_measure_and = set(card_x) & set(measure_x)
    card_and_measure = list(card_measure_and)
    
    if len(card_and_measure) >=3:
        print("成功")
        data.to_csv('TA2.csv', index=False, encoding='SHIFT_JIS')
    else:
        print("不可")

試したこと

空のリストの作製などしましたが，いまいち理解できず諦めてしまいました。

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

python 3.7, spyder

Answer 1

以下のようにbasenameで保存するとよいでしょう。
ちなみに以下コードでは読込元ファイルは元の場所にそのまま残ります。

import glob
import os
import pandas as pd

for file in glob.glob('TA*.csv'):
    basename = os.path.basename(file)
    df = pd.read_csv(file)
    if 条件を満たす:
        df.to_csv( os.path.join('/hoge/ok/', basename))
    else:
        df.to_csv( os.path.join('/hoge/ng/', basename))

Comments

[recrystal]

No such file or directory: '/hoge/ok\(ファイル名).csv
というエラーが出てしまいます。作業ディレクトリにフォルダを準備する必要があるのでしょうか。

[8524ba23]

はい。

[recrystal]

解決しました！

Answer 2

data.to_csv内で出力先のCSVファイル名がTA2.csv固定のなっている為、
同一ファイルに上書きされています。

for文で変数”file”にループ毎にCSVファイルが格納されているので

data.to_csv('TA2.csv', index=False, encoding='SHIFT_JIS')

↓

data.to_csv(file, index=False, encoding='SHIFT_JIS')

に修正してみてはいかがでしょうか。

Comments

[recrystal]

fileに変更することによって，すべてのファイルにプログラムを実行できたのですが，それらのファイルを二つのフォルダに振り分けることはできますでしょうか。

[K_Y_M]

os.path.join()
関数を用いれば、パスを指定する事が可能です。
pathok = ;
pathng = ;
等の変数で出力先を指定し、
if　条件式 :
data.to_csv(os.path.join(pathok,file), index=False, encoding='SHIFT_JIS')
else :
data.to_csv(os.path.join(pathng,file), index=False, encoding='SHIFT_JIS')

とすれば、出力先を分ける事は可能です。
詳細は以下のURLを参照ください。
https://www.sejuku.net/blog/64408

[recrystal]

解決しました，ありがとうございます