python上でのファイルやフォルダ情報を参照

よく使うけど毎回調べてしまうpythonのファイル操作関連のまとめ

フォルダ内のファイル一覧を取得

指定したフォルダ内のファイルの一覧をリストとして取得する

import glob

folder = "./tmp/" 
img_path_list = glob.glob(folder)

ワイルドカードを使用すれば特定のファイルの一覧も取得可能
下記はpngファイルのみをリストとして獲得

img_path_list = glob.glob(folder + "*.png")

ファイル名の取得

指定したパスからファイル名を文字列として取得する

import os

filepath = './tmp/hogehoge.ext'

filename = os.path.basename(filepath)
print(filename) # hogehoge.ext

拡張子なしで取得する場合

filename_only = os.path.splitext(os.path.basename(filepath))[0]
print(filename_only) # hogehoge 

存在しない場合にフォルダ作成

ファイルを保存するときになどに保存先フォルダが存在するかを確認し、
存在しない場合は新規に作成する

import os

filepath = "./hogehoge2/" 
if not os.path.exists(filepath):
        os.makedirs(filepath)