【Python】ファイルの拡張子を取得する

Pythonのファイル拡張子について。

今回は、Pythonでファイルの拡張子を調べる「endswith」の使い方を解説します。

ファイルの拡張子を取得する endswith

Pythonはファイルの拡張子を「endswith」で調べることができます。あるフォルダの中に入っている特定の拡張子のファイルだけ処理をしたい場合などで使えます。

endwith構文

文字列.endswith(末尾の文字列)

「endswith」では指定した文字列で終わっているか調べることができます。

戻り値

例1. 拡張子txtのファイルを取得する

import os
f_path = 'c:\\pg\\file'
for file in os.listdir(f_path):
    if file.endswith('.txt'):
        print(f_path + '\\' + file)
#[結果] c:\pg\file\file1.txt
#[結果] c:\pg\file\file2.txt

Pythonの「listdir」で「c:\pg\file」の中のファイルをすべて取得し、そのファイルの末尾が拡張子「.txt」で終わっているファイルのみを「endswith」を使って取得しました。

「endswith」は文字列の末尾を取得するので、拡張子以外にも文字列検索としても使えます。

パスの記述ですが、「\」をエスケープする必要があるため、「\」であれば「\\」と書きます。よって「c:\pg\file」の場合は「c:\\pg\\file」と書きます。

これで「endswith」で指定したフォルダ内のファイルの拡張子を調べることができました。

以上、Pythonでファイルの拡張子を取得する方法でした。