Python | ファイル名から拡張子を取得する方法

2023-02-12Python ファイル・ディレクトリ,Python

Pythonでファイル名からファイルの拡張子を取得する方法とサンプルコードを紹介しています。

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

Pythonではosモジュールの機能を使ってファイル名から拡張子を取得することが可能です。

拡張子を取得する部分のみのシンプルなコードにすると以下のように記述する事ができます。

import os

file = 'test.py'

ext = os.path.splitext(file)[1][1:] # ファイルの拡張子を取得
print(ext)

# py

指定したディレクトリのファイル一覧からファイルの拡張子を取得する

globモジュールを使って指定したディレクトリにあるファイル一覧を取得、それぞれ順番にファイルの拡張子を取得していきます。

glob()関数で取得したファイル名の一覧をfor文で反復してそれぞれ拡張子を取得を取得、処理していきます。

import glob
import os

files = glob.glob('dir/*.*') # ファイル一覧を取得

for value in files: # リストを繰り返す、valueにはファイル名が代入される
	ext = os.path.splitext(value)[1][1:] # ファイルの拡張子を取得
	# 処理
	print(ext)
	pass