Python;複数ファイル処理について

Question

前提・実現したいこと

python3.6を用いて処理を行おうと考えています．環境はwindowsです．
かなり初歩的な質問かと思いますが，ご容赦ください．

複数のファイルが同じディレクトリに存在し，そのひとつひとつに同様の処理を行います．
処理としては各ファイルにstftを行い，その結果を別のtxtファイルに出力するといったものです．

該当のソースコード

data=np.loadtxt("*.txt", delimiter='\n')
fs=2000

f, t, Zxx = signal.stft(data, fs, window=('gaussian',2),nperseg=216)
np.savetxt("*_stft.txt",Zxx, fmt='%.3f', delimiter='\n')

この*.txtの部分に次々とディレクトリ内のファイル名を入れ，処理していってほしいのですが，
どのようにすればよいのでしょうか．

Comments

[hayataka2049]

「二回目に実行したときに起こる悲劇」が容易に想像されるので、結果のファイルは拡張子を変えるか、別ディレクトリに置いた方が良いです。ご参考までに

[nn2]

アドバイスありがとうございます。拡張子を変えることにしました。

Answer 1

glob.globでカレントディレクトリのファイルを取得できます。

import glob

for filename in glob.glob('*.txt'):
    ...

保存先のファイル名は、filenameを加工して生成すれば良いです。

Comments

[nn2]

ありがとうございます。これを参考にさせてもらい、無事にプログラムができました。