非IT企業に勤める中年サラリーマンのIT日記

非IT企業でしかもITとは全く関係ない部署にいる中年エンジニア。唯一の趣味がプログラミングという”自称”プログラマー。

Python – フォルダ内のファイルリストを取得する

   

Python – フォルダ内のファイルリストを取得する

Pythonでフォルダ内のファイルリストを出力する必要があったので備忘録。osライブラリのlistdir関数を使います。

基本形

こちらが基本形です。

ファイルリスト(配列) = os.listdir(“パス”)

こちらのプログラムを実行すると指定したフォルダの中のファイルリストを出力してくれます。カレントディレクトリの場合はパスを ‘.’ と指定してあげます。

import os

files = os.listdir('.')
for file in files:
    print(file)
 

 

こちらが出力結果です。フォルダ内のすべてのファイル以外にフォルダもリストアップされています。

広告

フォルダを除外してファイルだけのリストを出力するには

フォルダを除外したい場合は、osライブラリのisfile関数でTrueになったものだけ出力すればOKです。

import os

files = os.listdir('.')
for file in files:
    if os.path.isfile(file):
        print(file)
 

 

出力結果がこちらです。出力からフォルダがなくなりました。

特定の拡張子だけのリストを出力するには

特定の拡張子だけのリストがほしい場合は、同じくosライブラリのsplitext関数を使って拡張子だけ抜き出せます。

以下のコードは.jpgの拡張子だけをリストアップする例です。

import os

files = os.listdir('.')
for file in files:
    if os.path.isfile(file):
        root, ext = os.path.splitext(file)
        if ext==".jpg":
            print(file)
 

 

以下の通り.jpgだけのリストになりました。

広告

 - Python