牌語備忘録 -pygo

あくまでもメモです。なるべくオフィシャルの情報を参照してください。

牌語備忘録 -pygo

特定の拡張子のついたファイルを取得してみた

#!/usr/bin/env python
# *-# -*- coding: utf-8 -*-

import os
import glob

#ひとつの拡張子のファイル名を取得したい場合
def hoge(dir_path, ext):
    os.chdir(dir_path)
    return glob.glob(ext)

#複数の拡張子のファイル名を取得したい場合
def fuga(dir_path, *ext_arr):
    os.chdir(dir_path)
    file_arr = []
    for ext in ext_arr:
        file_arr.extend(glob.glob(ext))
    return file_arr


if __name__ == '__main__':
    dir_path = '/Users/username/hoge/'
    print hoge(dir_path, '*.jpg')
    print fuga(dir_path, '*.jpg', '*.x3f')


もっとスマートな方法なかったっけ?