【Python】Pythonで特定の拡張子のファイルリストを作る

背景

• pythonでカレントディレクトリから特定の拡張子のファイルリストを作りたい時がある。

目標

• pythonでカレントディレクトリから特定の拡張子のファイルリストを作る。

コード

特定の拡張子だけのファイルリストの作り方

import os
files = os.listdir(os.getcwd())
md_files = [i for i in files if i.endswith('.md') == True]

特定の拡張子だけのファイルリストの作り方：コメント入り

import os

#カレントのファイルリストを読み込む
files = os.listdir(os.getcwd())

#特定の拡張子のファイルのみを取り出す
#（例として「.md」ファイルを取り出してる）
md_files = [i for i in files if i.endswith('.md') == True]

参考資料

個人ブログ