Python コードを 1 行ずつプロファイリングする

Hafiz Muhammad Zohaib 2024年2月15日
  1. プロファイリングとは
  2. line_profiler モジュールを使用した Python コードのプロファイリング
  3. コマンド ライン経由で Python コードをプロファイリングする
Python コードを 1 行ずつプロファイリングする

この記事では、Python コードを 1 行ずつプロファイリングし、コードの実行に関する有用な情報を取得する方法について説明します。

まず、プロファイリングについて簡単に紹介します。 次に、関数ベースのプロファイリングよりも行ごとのプロファイリングを使用する方が適している場合について説明します。 その後、Python でのプロファイリングの実装について説明します。

プロファイリングとは

プロファイリングは、コードのさまざまな部分で使用されるリソースをチェックするプロセスです。 効率的なコーディングの実践では、通常、時間の複雑さ (さまざまなコーディング ユニット/関数にかかる時間) またはメモリの複雑さ (さまざまなコーディング ユニット/関数によるメモリ使用量) に主に関心があります。

前者はプログラムにかかる時間を短縮することに集中するのに役立ち、後者はメモリ使用量を最適化するのに役立ちます。

関数プロファイリング

主にどの言語でも、関数レベルのプロファイリングは、プログラムのさまざまな関数によって消費される時間を調べるために使用されます。 したがって、Python の既定のパッケージには、関数プロファイリング用のライブラリ cProfileProfile が既に含まれています。

ラインプロファイラー

関数ベースのプロファイラーは、さまざまな関数にかかった時間に関する情報を提供します。 ただし、関数またはプログラムの全体的な時間にどの行がより多く貢献しているかを確認するために、行ベースの時間の複雑さが必要になる場合があります。

そのような場合のために、Python には line_profiler ライブラリがあります。

line_profiler モジュールを使用した Python コードのプロファイリング

LineProfiler モジュールを使用すると、コードを行ごとまたは関数ごとにプロファイリングできます。

次の Python コード スニペットでは、slow_avg_producer()fast_average_producer() という名前の 2つの関数を作成しました。 これらの関数はどちらも同じ平均を生成しますが、実行時間は異なります。

まず、次のモジュールをインストールして、ライン プロファイラーを使用してコードをプロファイリングします。

!pip install line_profiler
import time
import random


def slow_avg_producer():
    time.sleep(6)  # Do nothing for 5 time units
    # generate random numbers array
    arr = [
        random.randint(1, 100) for i in range(10000)
    ]  # generate random numbers array
    return sum(arr) / len(arr)


def fast_average_producer():
    time.sleep(1)  # Do nothing for 1 time units
    # generate random numbers array
    arr = [random.randint(1, 100) for i in range(10000)]
    return sum(arr) / len(arr)


def main_func():
    average = slow_avg_producer()
    print(average)

    result = fast_average_producer()
    print(average)

タイミング統計を取得するには、まず LineProfiler オブジェクトを作成し、次に main_func() を渡してラッパーを取得します。 最後に、プロファイリング統計を取得できます。

from line_profiler import LineProfiler

line_prof = LineProfiler()
line_prof_wrapper = line_prof(main_func)
line_prof_wrapper()
# print the profiling details
line_prof.print_stats()

上記のコードは、次の出力を生成します。 時間の単位はマイクロ秒です。

Timer unit: 1e-06 s

Total time: 7.10521 s
File: <ipython-input-14-67ae2a9633ee>
Function: main_func at line 17

Line #      Hits         Time  Per Hit   % Time  Line Contents
==============================================================
    17                                           def main_func():
    18         1    6054801.0 6054801.0     85.2      average = slow_avg_producer()
    19         1        676.0    676.0      0.0      print(average)
    20
    21         1    1049070.0 1049070.0     14.8      result = fast_average_producer()
    22         1        659.0    659.0      0.0      print(average)

統計によると、メイン関数では、slow_avg_producer() に 6.054 秒かかり、次の print ステートメントに 0.000676 秒かかります。 一方、fast_average_producer() は非常に短い時間しかかかりません。

コマンド ライン経由で Python コードをプロファイリングする

コマンド ライン インターフェイスを使用して、関数の行ごとのプロファイリングの詳細を取得できます。 この方法では、kernprof コマンドを使用します。

ここでも同じ関数を使用しますが、各関数の先頭にデコレータ @profile を追加します。 このスクリプトに average_producer.py という名前を付けます。

import time
import random


@profile
def slow_avg_producer():
    time.sleep(6)  # Do nothing for 5 time units
    # generate random numbers array
    arr = [
        random.randint(1, 100) for i in range(10000)
    ]  # generate random numbers array
    return sum(arr) / len(arr)


@profile
def fast_average_producer():
    time.sleep(1)  # Do nothing for 1 time units
    # generate random numbers array
    arr = [random.randint(1, 100) for i in range(10000)]
    return sum(arr) / len(arr)


@profile
def main_func():
    average = slow_avg_producer()
    print(average)

    result = fast_average_producer()
    print(average)


main_func()

kernprof コマンドを使用して、次のコマンドを使用して line-profiler モジュールをインストールする必要があります。

pip install line-profiler

Colab または Jupyter Notebook ターミナルの場合、上記のコマンドの先頭に感嘆符 (!) を使用します。

line-profiler をインストール

インストール後、kernprof.exe がインストールされているディレクトリを変更する必要があります。 次に、次のコマンドを実行します。

kernprofcProfile を実行するオプションを提供しますが、ここではライン プロファイラーを使用します。 そのために、-l 引数が使用されます。

kernprof -l average_producer.py

kernprof

次のコマンドを実行して、プロファイリングの結果を確認します。

python -m line_profiler average_producer.py.lprof

以下に示すように、プロファイリング結果をコマンド ライン インターフェイスに出力します。

python ライン プロファイラー

プロファイリング結果

関連記事 - Python Profile