カテゴリー: python

ディレクトリ直下のファイル一覧、ディレクトリ一覧を３つの方法で取得します。

ディレクトリは以下、ファイルはtxtとlogの２種類です。

ディレクトリ、ファイル構成

sample_dir
  │  
  │  f_01.txt
  │  f_02.log
  │
  ├─d_01
  │      f_01_1.txt
  │      f_01_2.txt
  │      f_01_3.txt
  │
  └─d_02
          f_02_1.log

pathlibで ディレクトリ配下のファイル一覧を取得する

pathlib.Pathを利用して、 拡張子を指定したファイル一覧およびディレクトリとファイル一覧を表示する サンプルコード です。

import pathlib

p = pathlib.Path("sample_dir")

# ディレクトリ配下のテキストファイル一覧を表示（拡張子指定）
for f in p.glob("**/*.txt"):
    print(f)

print()

# ディレクトリ配下のディレクトリ、ファイル一覧を表示
for f in p.glob("**/*"):
    print(f)


# - 結果1
# sample_dir\f_01.txt
# sample_dir\d_01\f_01_1.txt
# sample_dir\d_01\f_01_2.txt
# sample_dir\d_01\f_01_3.txt

# - 結果2
# sample_dir\d_01
# sample_dir\d_02
# sample_dir\f_01.txt
# sample_dir\f_02.log
# sample_dir\d_01\f_01_1.txt
# sample_dir\d_01\f_01_2.txt
# sample_dir\d_01\f_01_3.txt
# sample_dir\d_02\f_02_1.log

pathlib.Path関数にディレクトリのパスを指定します。glob関数で取得するサンプルは相対パスですが、絶対パスも指定可能です。

pathlibの詳細については、公式ドキュメントを参照ください。

globでディレクトリ配下のファイル一覧を取得する

globを利用して、拡張子を指定したファイル一覧およびディレクトリとファイル一覧を表示する サンプルコード です。

import glob

TARGET_DIR = "sample_dir"

# ディレクトリ配下のテキストファイル一覧を表示（拡張子指定）
for f in glob.glob(TARGET_DIR + "/**/*.txt", recursive=True):
    print(f)

print()

# ディレクトリ配下のディレクトリ、ファイル一覧を表示
for f in glob.glob(TARGET_DIR + "/**", recursive=True):
    print(f)


# - 結果1
# sample_dir\f_01.txt
# sample_dir\d_01\f_01_1.txt
# sample_dir\d_01\f_01_2.txt
# sample_dir\d_01\f_01_3.txt

# - 結果2
# sample_dir\
# sample_dir\d_01
# sample_dir\d_01\f_01_1.txt
# sample_dir\d_01\f_01_2.txt
# sample_dir\d_01\f_01_3.txt
# sample_dir\d_02
# sample_dir\d_02\f_02_1.log
# sample_dir\f_01.txt
# sample_dir\f_02.log

glob.glob関数にパスを指定します、再帰的に検索する行うには、「recursive」に「True」を指定します。

globの詳細については、公式ドキュメントを参照ください。

walkでディレクトリ配下のファイル一覧を取得する

os.walkを利用して、拡張子を指定したファイル一覧およびディレクトリとファイル一覧を表示する サンプルコード です。

import os

TARGET_DIR = "sample_dir"

# ディレクトリ配下のテキストファイル一覧を表示
for dir_path, dir_names, file_names in os.walk(TARGET_DIR):
    for file_name in file_names:
        if ".txt" in file_name:
            print(os.path.join(dir_path, file_name))

print()

# ディレクトリ配下のディレクトリ、ファイル一覧を表示
for dir_path, dir_names, file_names in os.walk(TARGET_DIR):
    for dir_name in dir_names:
        print(os.path.join(dir_path, dir_name))
    for file_name in file_names:
        f = os.path.join(dir_path, file_name)
        print(f)


# - 結果1
# sample_dir\f_01.txt
# sample_dir\d_01\f_01_1.txt
# sample_dir\d_01\f_01_2.txt
# sample_dir\d_01\f_01_3.txt

# - 結果2
# sample_dir\d_01
# sample_dir\d_02
# sample_dir\f_01.txt
# sample_dir\f_02.log
# sample_dir\d_01\f_01_1.txt
# sample_dir\d_01\f_01_2.txt
# sample_dir\d_01\f_01_3.txt
# sample_dir\d_02\f_02_1.log

os.walk関数にパスを指定します。これまでと違うのは、os.walkでは、ディレクトリパス、ディレクトリ名、ファイル名の３つの戻り値が返却されます。

os.walkの詳細については、公式ドキュメントを参照ください。

型の確認方法と型チェックをする記載例です。

Python 型を確認する

type(object)

例

type(1) -> <class ‘int’>

Python 型をチェックする

isinstance(object, class)

例

isinstance(1, int) -> True

Python 型を確認、チェックする サンプルコード

# int
i = 1
print(type(i))
print(isinstance(i, int))

# str
s = "a"
print(type(s))
print(isinstance(s, str))

# list
l = ["a", 1]
print(type(l))
print(isinstance(l, list))

# dict
d = {"key": "value"}
print(type(d))
print(isinstance(d, dict))

# tuple
t = (1, 2)
print(type(t))
print(isinstance(t, tuple))

print()

# int check.int or listなら True.
print(isinstance(i, (int, list)))
# str check.int or listなら True.
print(isinstance(s, (int, list)))
# list check.int or listなら True.
print(isinstance(l, (int, list)))
# dict check.int or listなら True.
print(isinstance(d, (int, list)))


# ===== 出力例 =====
# <class 'int'>
# True
# <class 'str'>
# True
# <class 'list'>
# True
# <class 'dict'>
# True
# <class 'tuple'>
# True

# True
# False
# True
# False

print関数のデフォルト設定では、出力文字の末尾に改行コードが設定されます。endパラメータを追加することで、変更できます。

Printを改行なしにする

print(“hoge”, end=””)

end=””（空文字）を設定します

Printの区切り文字をなしにする

print(“foo”, “bar”, sep=””)

sep=””（空文字）を設定します

サンプルコード

# デフォルトは末尾が改行コード
print("line_01")
print("line_02")
# 改行なし
print("line_03", end="")
print("line_04")
# カンマ
print("line_05", end=",")
print("line_06")

print()

# 文字列複数
print("foo", "bar")
# 区切り文字をなし
print("foo", "bar", sep="")

# ===== 出力例 =====
# line_01
# line_02
# line_03line_04
# line_05,line_06

# foo bar
# foobar

よく使うので、メモ。

Pythonでディレクトリの存在チェックして、存在しなければ作成する

ディレクトリの存在チェックと作成するサンプルコード

import os

SAMPLE_DIR = "sample"

if not os.path.exists(SAMPLE_DIR):
    # ディレクトリが存在しない場合、ディレクトリを作成する
    os.makedirs(SAMPLE_DIR)

os.path.exists関数にディレクトリのパスを渡します。ディレクトリが存在する場合は、True。存在しない場合は、Falseが返却されます。

os.makedirs関数にディレクトリのパスを渡してディレクトリを作成します。

※上記の記述はファイルが存在した場合も、Trueとなります。

長めのサンプルコード

import os

SAMPLE_DIR = "sample2"


# 存在チェック
if os.path.isdir(SAMPLE_DIR):
    print("ディレクトリが存在します")
else:
    print("ディレクトリが存在しません")

# ディレクトリがない場合、作成する
if not os.path.exists(SAMPLE_DIR):
    print("ディレクトリを作成します")
    os.makedirs(SAMPLE_DIR)

# 存在チェック
if os.path.isdir(SAMPLE_DIR):
    print("ディレクトリが存在します")
else:
    print("ディレクトリが存在しません")

長めのサンプルコードでは、ディレクトリの存在チェックにos.path.isdirを利用しています。この関数を利用すれば、ファイルの場合はFalseが帰ってきます。

実行例

>>python check_exists_dir.py

ディレクトリが存在しません
ディレクトリを作成します
ディレクトリが存在します

日付、時刻をよく利用することがあるので、メモとして掲載します。datetimeとPOSIX タイムスタンプ(UnixTime)を相互変換するサンプルコードです。

UTC⇔JSTの変換サンプルコード

解説

各処理の解説です。

利用するモジュールを宣言します。

from datetime import datetime, timedelta, timezone

それぞれ日付を扱うために利用するクラスであり、datetimeは日時を扱い、timedeltaは経過時間（日付や時刻間の差）を扱い、timezoneは、タイムゾーンを扱います。

UTCからJSTを計算する

# 現在時刻（UTC）をdatetime型で取得する
dt_utc = datetime.now(timezone.utc)
print("UTC        : {}, type : {}".format(dt_utc, type(dt_utc)))

# UTC -> JST
JST = timezone(timedelta(hours=+9))
utc_jst = dt_utc.astimezone(JST)
print("UTC -> JST : {}, type : {}".format(utc_jst, type(utc_jst)))

datetime.now()に引数としてtimezone.utcを指定してdatetime型のUTC日時を取得します。 timezone.utc を指定しない場合、ローカル時間（プログラムが実行されているパソコンやサーバーの設定時間）が取得されます。

UTCからJSTへの変換には、datatime.astimezone()にUTCとJSTの時差+9を追加したタイムゾーン（サンプルの場合、JST）を指定します。UTCはデフォルトでtimezone.utcに定義されていますが、JSTは自分で定義する必要があります。

JSTからUTCを計算する

jst_utc = utc_jst.astimezone(timezone.utc)
print("JST -> UTC : {}, type : {}".format(dt_utc, type(dt_utc)))

JSTからUTCを指定する際にも、datetime.astimezone()へUTCのタイムゾーンを指定します。

JSTを取得する

# 現在時刻（JST）をdatetime型で取得する
JST = timezone(timedelta(hours=+9))
dt_jst = datetime.now(JST)
print("JST        : {}, type : {}".format(dt_jst, type(dt_jst)))

datetime.now()へJSTのタイムゾーンを指定します。

datetime -> POSIXタイムスタンプ（UnixTIme）とその逆

# 現在時刻（UTC）をPOSIXタイムスタンプ(float型)で取得する
ts_utc = dt_utc.timestamp()
print("timestamp utc : {}, type : {}".format(ts_utc, type(ts_utc)))

# 現在時刻（JST）をPOSIXタイムスタンプ(float型)で取得する
ts_jst = dt_jst.timestamp()
print("timestamp jst : {}, type : {}".format(ts_jst, type(ts_jst)))

# POSIXタイムスタンプ(float型)をもう一度datetimeに変換する
dt_jst_2 = datetime.fromtimestamp(ts_jst, JST)
print("timestamp -> datatime2 : {}".format(dt_jst_2))
print("datatime == datatime2  : {}".format(dt_jst == dt_jst_2))

datetime.timestamp()でdatetimeをPOSIXタイムスタンプに変換します。

POSIXタイムスタンプをdatetimeに変換するには、timestamp()を使います。