pythonで指定拡張子を一括移動

colabを使っていて、拡張子がpngやjpgといった画像ファイルを一括で移動させたかったので、書いてみました。

move.py

import os
import shutil
import re

DIR = 'moved_dir' # 移動後のディレクトリ
EXT_LIST = ['.png',".PNG",".jpg"] # 移動させたい拡張子のリスト

# 移動後のディレクトリがなかったら作成
if not os.path.isdir(DIR):
    os.mkdir(DIR)

# 上のディレクトリにファイルを移動
for file_name in os.listdir(): # 現在いる場所のファイルやディレクトリをリストで取得（もっと効率のいい方法がありそう）
    for ext in EXT_LIST:
        m = re.search(r'.*' + ext, file_name) 
        if m != None:
          shutil.move(m.group(), INPUTDIR) # 各々の拡張子にマッチするファイル名があったら、移動させる

globのノリでワイルドカード使ったら、shutil.moveでは使えないんですね。

以上、個人的備忘録でした。