Pythonでos.listdirを使った順番通りの再帰処理をマスター! ｜自作で機械学習モデル・AIの使い方を学ぶ

今回はPythonのosモジュールを使った再帰処理について解説します。

Pythonのosモジュールについて

Pythonのosモジュールは、オペレーティングシステムとやり取りするためのモジュールです。

ファイルやディレクトリの作成、コピー、移動、削除などを行うことができます。

また、osモジュールを使ってファイルやディレクトリの一覧を取得することもできます。

os.listdirの基本的な使い方

os.listdirは、指定したディレクトリ内のファイルやディレクトリの一覧を取得するための関数です。

以下のように使用します。

import os
files = os.listdir('/path/to/directory')
print(files)

上記のプログラムでは、/path/to/directoryディレクトリ内のファイルやディレクトリの一覧を取得して、変数filesに格納しています。

取得した一覧はリスト形式で返されます。

os.listdirを用いたディレクトリの一覧表示

os.listdirを使って、指定したディレクトリ内のファイルやディレクトリの一覧を表示するプログラムを作成してみましょう。

import os
def list_files(directory):
    files = os.listdir(directory)
    for file in files:
        print(file)
list_files('/path/to/directory')

上記のプログラムでは、指定したディレクトリ内のファイルやディレクトリの一覧を、forループを使って順番に表示しています。

os.listdirを使った再帰処理の方法

os.listdirを使って、指定したディレクトリ内のファイルやディレクトリの一覧を再帰的に取得するプログラムを作成してみましょう。

import os
def list_files_recursively(directory):
    for root, dirs, files in os.walk(directory):
        for file in files:
            print(os.path.join(root, file))
list_files_recursively('/path/to/directory')

上記のプログラムでは、os.walkを使って指定したディレクトリ内のファイルやディレクトリの一覧を再帰的に取得しています。

os.walkは、指定したディレクトリ以下のファイルやディレクトリを再帰的に探索するための関数です。

os.walkは、ルートディレクトリ、サブディレクトリ、ファイルの3つの要素を順番に返します。

os.listdirでのファイル・ディレクトリの順番通りの表示方法

os.listdirで取得したファイルやディレクトリの一覧を、順番通りに表示するためには、以下のようにsort関数を使ってソートしてから表示します。

import os
def list_files(directory):
    files = os.listdir(directory)
    files.sort()
    for file in files:
        print(file)
list_files('/path/to/directory')

上記のプログラムでは、os.listdirで取得したファイルやディレクトリの一覧をsort関数を使ってソートしてから、順番通りに表示しています。

os.listdirを使った順番通りの再帰処理の実装例

os.listdirを使って、指定したディレクトリ内のファイルやディレクトリを順番通りに再帰的に取得するプログラムを作成してみましょう。

import os
def list_files_recursively(directory):
    files = os.listdir(directory)
    files.sort()
    for file in files:
        full_path = os.path.join(directory, file)
        if os.path.isdir(full_path):
            list_files_recursively(full_path)
        else:
            print(full_path)
list_files_recursively('/path/to/directory')

上記のプログラムでは、os.listdirで取得したファイルやディレクトリの一覧をsort関数を使ってソートしてから、順番通りに再帰的に取得しています。

また、取得した要素がディレクトリの場合には、再帰的に取得するために、list_files_recursively関数を再帰的に呼び出しています。