【Python】ファイルあれこれ

良く忘れるので、備忘録的にPythonのファイル操作関連。

まずは、ファイル一覧の取得。ファイル名とディレクトリ名の一覧の取得。カレントディレクトリで試してみます。

import os

path = "./"
files = os.listdir(path)

print(files)

['sys_mod.py', 'folder1', 'folder2']

ディレクトリのみ取得。

import os

path = "./"
files = os.listdir(path)
files_dir = [f for f in files if os.path.isdir(os.path.join(path, f))]

print(files_dir)

['folder1', 'folder2']

標準モジュールのglobを使ってもOK。

import glob

files = glob.glob('./*')

print(files)

print(type(files))

['.\\sys_mod.py', '.\\folder1', '.\\folder2']
<class 'list'>

ファイルサイズの取得

import os

path = "./"
files = os.listdir(path)

print(files)

for f in files:
   print(os.path.getsize(f))

['sys_mod.py', 'folder1', 'folder2']
196
0
0

ディレクトリが存在するかどうか。

import os

path = "./"
files = os.listdir(path)

print(files)

for f in files:
   print(os.path.getsize(f))
   if os.path.isdir(f):
       print("Directory exists")

['sys_mod.py', 'folder1', 'folder2']
255
0
Directory exists
0
Directory exists

すぐ忘れるんだよなあ。