Python で walk 関数を使ってパス以下のファイル・ディレクトリ一覧を取得して表示

phiary に引っ越しました. 毎日プログラミングやWebに関する情報を発信しています! RSS 登録してたまに覗いたり, tweet やハテブして拡散してもらえると幸いです.

Pocket

Python で walk 関数を使ってパス以下のファイル・ディレクトリ一覧を取得して表示

# coding: utf-8
# パス以下のファイルの一覧を取得して表示

import os

# ----------------------------------------------------------------
# ファイルをリストとして取得
# ----------------------------------------------------------------
def getFileList(path):
    file_list = []
    for (root, dirs, files) in os.walk(path):
        for file in files:
            # パスセパレータは\\より/の方が好きなので置換
            file_list.append( os.path.join(root,file).replace("\\", "/") )
        
    return file_list

# ----------------------------------------------------------------
# ディレクトリをリストとして取得
# ----------------------------------------------------------------
def getDirList(path):
    dir_list = []
    for (root, dirs, files) in os.walk(path):
        for dir in dirs:
            # パスセパレータは\\より/の方が好きなので置換
            dir_list.append( os.path.join(root,dir).replace("\\", "/") )
        
    return dir_list

# ================================================================
# main時のtest
# ================================================================
def main():
    # ファイルを全て表示
    print("<file list>")
    for f in getFileList("C:/Python32"): print(f)
    # ディレクトリを全て表示
    print("<directory list>")
    for d in getDirList("C:/Python32"): print(d)

# メイン時
if __name__ == '__main__': main()

TRACK BACK URL

POST COMMENT

メールアドレスが公開されることはありません。