Pythonで等差（共通の差分）を持つ行だけを抽出するプログラムの書き方

リストの中から「隣り合う要素同士の差が一定（等差数列）になっている行」だけを抽出したい場合、繰り返し処理とフラグ変数を組み合わせるのが基本的なアプローチです。

サンプルコード

以下に具体的な実装例を示します。

my_list = [[31, 27, 10], [8, 11, 12], [11, 12, 13], [6, 9, 10]]

print("元のリスト：")
print(my_list)

my_result = []

for row in my_list:
    temp = True
    for index in range(0, len(row) - 1):

        if row[index + 1] - row[index] != row[1] - row[0]:
            temp = False
            break

    if temp:
        my_result.append(row)

print("抽出結果のリスト：")
print(my_result)

実行結果

元のリスト：
[[31, 27, 10], [8, 11, 12], [11, 12, 13], [6, 9, 10]]
抽出結果のリスト：
[[11, 12, 13]]

処理の流れと解説

• まず、複数のリスト（行）を含むネストされたリストを定義し、コンソールに表示します。
• 抽出結果を格納するための空のリストを作成します。
• 外側のループで各行を順番に取り出し、判定用のフラグ変数を True で初期化します。
• 内側のループで、行内の各要素のインデックスを走査します。
• 隣接する要素同士の差が、先頭2要素の差（基準となる公差）と一致しない場合、フラグ変数を False に設定します。
• 不一致が見つかった時点で break により内側のループを抜け、無駄な比較を省きます。
• 最終的にフラグが True のままなら、その行は等差数列であるため、結果用のリストに追加します。
• すべての行の判定が完了したら、結果のリストをコンソールに出力します。

ポイント

この手法では、各行の先頭2要素の差 row[1] - row[0] を基準値として扱い、以降のすべての隣接要素間の差がこの基準と一致するかどうかを検証しています。これにより、[11, 12, 13] のように公差が一定の行だけが正確に抽出されます。要素数が2以下の行は常に条件を満たす点にも注意してください。