【python3】フォルダ内のPDFファイルをリストにまとめるコード。

フォルダ内のある種類のファイルに対して処理をする際に使えるリストアップ部分のすっきりしたコードを見つけたので残しておきます。
今回はpdfですが、もちろんjpegでもcsvでもOKです。

import os


pdf_files = []
for filename in os.listdir('.'):
    if filename.endswith('.pdf'):
        pdf_files.append(filename)
print(pdf_files)

import os

pdf_files = []

for filename in os.listdir('.'):

if filename.endswith('.pdf'):

pdf_files.append(filename)

print(pdf_files)

‘.pdf’の部分を探したい拡張子に変更することで指定のファイルをリスト形式でまとめることが出来ます。