python;複数ファイルを次々に処理

Question

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

複数のファイルが同じディレクトリに存在し，そのひとつひとつに同様の処理を行います．
処理としては，replaceを行い，文字列置換された結果を別ファイルとして出力(書き出し)するというものです．

f=open('hoge.txt')
data=f.read()
f.close()
data2=data.replace("\t",",")
n=open('hoge.csv','w')
n.write(data2)
n.close()

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

Answer 1

Python3.4とのことなので、pathlibを使うと結構楽に書けると思います。

import sys
from pathlib import Path

path = Path(sys.argv[1] if len(sys.argv) >= 2 else '.')

for path_in in [x for x in path.glob('*.txt') if x.is_file()]:
    path_out = path_in.with_suffix('.csv')
    print('{} -> {} ...'.format(path_in, path_out))
    with path_in.open() as fin, path_out.open('w') as fout:
        fout.write(fin.read().replace('\t', ','))

ただし、pathlibは暫定パッケージなので将来にわたって使える保証がありません。それを気にする場合はosモジュールのlistdir()を使うことになります。pathlibに比べるとちょっとだけ冗長ですが。

import os
import sys

target_dir = sys.argv[1] if len(sys.argv) >= 2 else '.'

for filename in os.listdir(target_dir):
    path_in = os.path.join(target_dir, filename)
    if not (os.path.isfile(path_in) and path_in.endswith('.txt')):
        continue
    path_out = path_in.replace('.txt', '.csv')
    print('{} -> {} ...'.format(path_in, path_out))
    with open(path_in) as fin, open(path_out, 'w') as fout:
        fout.write(fin.read().replace('\t', ','))

WindowsでPython、とのことなのでこういう回答にしましたが、もしPosix準拠(UNIX系)のOSならfind、xargs、sedなどの組み合わせ技でやりましょう、と言っているところです。WindowsでもMSYS2やCygwinでPosix環境が試せるので興味があったら調べてみてください。

Answer 2

2 つの *.txt ファイルと 1 つの python ファイルを用意してみました。
1.txt

1	2	3
あ	い	う
"123"	"あいう"	""

2.txt

a	b	c
１２３	４５６	７６８
abc	あいう	aあ

1.py

import os
import glob

# in_filname の内容を変換して、out_filename に出力する。
def raplace_file(in_filename, out_filename):
  # print(in_filename + " => " + out_filename)
  lines = []
  fin = open(in_filename, 'r')
  lines = fin.readlines() # 1行毎にファイル終端まで全て読む(改行文字も含まれる)
  fin.close()

  # 必要なら上書きチェックや出力ファイル名の変更などの処理を追加する事。
  fout = open(out_filename, 'w')
  for line in lines:
    # \t を , に置き換える。
    fout.write(line.replace("\t",","))
  fout.close()

directory = os.getcwd()
for filename in glob.glob(directory + '/*.txt'):
  path, _ = os.path.splitext(filename)
  raplace_file(filename, path + ".csv")

実行例：

$ ls
1.py	1.txt	2.txt

$ python3 1.py

$ ls
1.csv	1.py	1.txt	2.csv	2.txt

$ cat 1.csv
1,2,3
あ,い,う
"123","あいう",""

$ cat 2.csv
a,b,c
１２３,４５６,７６８
abc,あいう,aあ