globモジュールを使ってローカルに存在するファイル群を取得する方法を記載します。
globモジュールとは
Unixシェルが使用する規則に従って、指定されたパターンに合致するパス名をすべて見つけることができます。
使い方
まずglobをimportします。glob自体は最初から含まれているため、pipでインストールする必要はありません。
import glob
.pdfと記載されたファイルを取得するには、以下のように記載します。
import glob
glob.glob('*.pdf')
出力結果は以下です。
['1.pdf', '3.pdf', '2.pdf']
.pdfとついたファイルが全て取得できました。
リスト型で取得されるため、1つずつ処理したい場合は、for文で処理する必要があります。
import glob
files = glob.glob('*.pdf')
for file in files:
print(file)
以下が出力されます。
1.pdf
3.pdf
2.pdf
また、取得するパスは特に指定がなければ、カレントディレクトリから取得されます。
また、*のような特殊文字は*だけでなく、?なども使用できます。
公式ドキュメント
公式ドキュメントはリンクを貼っておきますので、詳細は以下リンクを参照するようお願いします。
https://docs.python.org/ja/3/library/glob.html