Pythonのitertoolsを用いて与えられたリストの要素のペアを取得し表示する

スポンサーリンク

Pythonのitertoolsを用いて与えられたリストの要素のペアを取得し表示してみます。

今回はitertoolsモジュールを用います。このモジュール・ライブラリはPythonの標準ライブラリですので、事前にインストールする必要はありません。

■Python

今回のPythonのバージョンは、「3.10.9」を使用しています。(Windows11)(pythonランチャーでの確認)

■itertoolsを用いて与えられたリストの要素のペアを取得し表示する

では、itertoolsを用いて与えられたリストの要素のペアを取得し表示するスクリプトを書いていきます。

■コード

from itertools import pairwise

def print_pairs(lst):
    # pairwise関数を使ってリスト内の要素のペアを取得
    pairs = list(pairwise(lst))

    # ペアを表示
    for pair in pairs:
        print(pair)

# テスト用のリスト
my_list = [1, 2, 3, 4, 5]

# ペアを表示する
print_pairs(my_list)

まずは、itertoolsモジュールからpairwise関数をインポートします。pairwise関数は、リスト内の要素のペアを生成するイテレータを作成する関数です。

次にprint_pairsという関数を定義します。この関数は、リスト内の要素のペアを取得し、それらのペアを表示する役割を持ちます。その後、pairwise関数を使ったペアの取得を行います。pairwise(lst)を使って、与えられたリストlst内の要素のペアを取得します。pairwise関数はイテレータを返すため、直接list()関数を使ってイテレータをリストに変換しています。

変換後、ペアの表示します。取得したペアをforループを使用して一つずつ表示します。ループ内でpairという変数にペアが代入され、それをprint()関数を使って表示します。

次にテスト用のリストを作成します。今回は「my_list = [1, 2, 3, 4, 5]」というリストを作成。このリストは、テスト用のデータとしてprint_pairs関数に渡されます。

作成後、print_pairs関数の呼び出します。print_pairs(my_list)を使って、先ほど作成したmy_listをprint_pairs関数に渡し、ペアを表示します。

■実行・検証

このスクリプトを「i_p_test.py」という名前で、Pythonが実行されている作業ディレクトリ(カレントディレクトリ)に保存し、コマンドプロンプトから実行してみます。

実行してみると、pairwise関数は隣接する要素のペアを生成するため、(1, 2)、(2, 3)、(3, 4)、(4, 5)のように隣り合った要素がペアとして表示されることを確認できました。

コメント

タイトルとURLをコピーしました