Python – K番目のセグメントごとにクロス結合（交互マージ）する方法

2つのリストの要素を「K」個ずつ交互に組み合わせる（クロス結合する）処理が必要な場合があります。そんなときは、ジェネレータ関数を使ってインデックスを追跡しながら要素を順番に取り出すメソッドを定義すると、シンプルかつ効率的に実装できます。

本記事では、PythonでK番目のセグメント単位に2つのリストを交互にマージする方法を、具体的なコード例とともに解説します。

サンプルコード

以下は、2つのリストをK個ずつ交互に結合する実装例です。

def merge_pair_elem(my_list_1, my_list_2, K):
   index_1 = 0
   index_2 = 0
   while(index_1 < len(my_list_1)):

      for i in range(K):
         yield my_list_1[index_1]
         index_1 += 1
      for i in range(K):
         yield my_list_2[index_2]
         index_2 += 1

my_list_1 = [24, 13, 82, 22, 65, 74]
my_list_2 = [55, 63, 17, 44, 33, 15]

print("The first list is :")
print(my_list_1)

print("The second list is :")
print(my_list_2)
K = 1

print("The value of K is :")
print(K)

my_result = [element for element in merge_pair_elem(my_list_1, my_list_2, K)]

print("The result is :")
print(my_result)

出力結果

The first list is :
[24, 13, 82, 22, 65, 74]
The second list is :
[55, 63, 17, 44, 33, 15]
The value of K is :
1
The result is :
[24, 55, 13, 63, 82, 17, 22, 44, 65, 33, 74, 15]

コードの解説

• merge_pair_elem という名前のメソッドを定義します。このメソッドは、2つのリストと「K」の値を引数として受け取ります。

• メソッド内部では、各リスト用のインデックス変数（index_1、index_2）を0で初期化し、whileループの中で yield を使って要素を1つずつ生成します。これにより、最初のリストからK個、次に2つ目のリストからK個という順序で要素が交互に取り出されます。

• メソッドの外側では、整数のリストを2つ定義し、コンソールに表示しています。

• 交互に結合する単位となる「K」の値を定義し、こちらもコンソールに表示します。

• リスト内包表記を使ってジェネレータを反復処理し、必要なパラメータを渡してメソッドを呼び出します。

• 生成された要素はリストに変換され、変数 my_result に代入されます。

• 最後に、その結果がコンソールに出力されます。

補足：ジェネレータを使うメリット

この実装では yield を使ったジェネレータ関数を採用しています。ジェネレータは要素を一度にすべてメモリ上に展開するのではなく、必要なタイミングで1つずつ値を生成するため、大量のデータを扱う場合でもメモリ効率に優れているのが特徴です。また、Kの値を変更するだけで「2個ずつ」「3個ずつ」といった異なるパターンの交互結合にも柔軟に対応できます。