Pythonでファイル拡張子を抽出する方法【os.path.splitextの使い方】

Pythonでは、標準ライブラリの os.path.splitext メソッドを使うことで、ファイル名の文字列から拡張子を簡単に抽出できます。

このメソッドは、パス名を (root, ext) のペア（タプル）に分割します。分割後も root + ext == path の関係が成り立ちます。また、ext は空文字列であるか、ピリオド（.）で始まり、最大1つのピリオドのみを含む文字列になります。

基本的な使い方

以下は、ファイル名から拡張子を抽出するシンプルな例です。

import os

file_name = 'my_file.txt'
print(os.path.splitext(file_name))

実行結果

('my_file', '.txt')

このように、ファイル名部分と拡張子部分がタプルとして返されます。拡張子だけを取り出したい場合は、インデックス [1] を指定します。

import os

file_name = 'my_file.txt'
extension = os.path.splitext(file_name)[1]
print(extension)  # 出力: .txt

補足： pathlib を使う方法

Python 3.4以降では、よりモダンな pathlib.Path.suffix を使う方法もあります。

from pathlib import Path

file_name = Path('my_file.txt')
print(file_name.suffix)  # 出力: .txt

どちらの方法でも拡張子を取得できますが、パス操作全般を行う場合は pathlib の利用が推奨されています。用途に応じて使い分けるとよいでしょう。