【Python】ファイル操作あれこれ

Pythonは、簡単に扱えるだけではなく、Webアプリ作成やデータサイエンスの分野でも広く使われています。

様々な場面でファイル操作が必要になりますが、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

これらの操作は、使用頻度がそれほど多くないためか、良く忘れるんですよねえ。

備忘録的に書いておきました。