Pythonでプレフィックス（接頭辞）の出現位置を基準に文字列リストを分割する方法

Pythonでは、特定のプレフィックス（接頭辞）が出現する位置を基準に文字列のリストを分割したいケースがあります。このような場合、まず2つの空のリストを定義し、基準となるプレフィックスの値を設定します。その後、シンプルな反復処理とappendメソッドを組み合わせることで、目的の分割を実現できます。

サンプルコード

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

from itertools import zip_longest

my_list = ["hi", 'hello', 'there', "python", "object", "oriented", "object", "cool", "language", 'py', 'extension', 'bjarne']

print("The list is : ")
print(my_list)

my_prefix = "python"
print("The prefix is :")
print(my_prefix)

my_result, my_temp_val = [], []

for x, y in zip_longest(my_list, my_list[1:]):
   my_temp_val.append(x)
   if y and y.startswith(my_prefix):
      my_result.append(my_temp_val)
      my_temp_val = []

my_result.append(my_temp_val)

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

print("The list after sorting is : ")
my_result.sort()
print(my_result)

出力結果

The list is :
['hi', 'hello', 'there', 'python', 'object', 'oriented', 'object', 'cool', 'language', 'py', 'extension', 'bjarne']
The prefix is :
python
The resultant is :
[['hi', 'hello', 'there'], ['python', 'object', 'oriented', 'object', 'cool', 'language', 'py', 'extension', 'bjarne']]
The list after sorting is :
[['hi', 'hello', 'there'], ['python', 'object', 'oriented', 'object', 'cool', 'language', 'py', 'extension', 'bjarne']]

コードの解説

• まず、必要なパッケージ（itertoolsモジュール）を環境にインポートします。

• 文字列要素を持つリストを定義し、その内容をコンソールに表示します。

• 分割の基準となるプレフィックスの値を定義し、こちらもコンソールに表示します。

• 結果を格納するための空のリストを2つ用意します。1つは最終結果用、もう1つは一時的なグループ保持用です。

• zip_longestメソッドを使用して、元のリストと「先頭の1要素を除外した同じリスト」をペアにしながら反復処理を行います。これにより、現在の要素と次の要素を同時に参照できます。

• 各反復処理の中で、現在の要素を一時リストに追加します。そして次の要素が存在し、かつその要素が指定したプレフィックスで始まる場合、一時リストを結果リストに追加したうえで、一時リストを初期化します。

• ループ終了後、残った一時リストも結果リストに追加することで、分割処理が完了します。

• 最後に、得られたリストをsortメソッドでソートし、コンソールに表示します。

ポイント：zip_longestについて

zip_longestは、長さの異なる複数のイテラブルを組み合わせる際に便利な関数です。通常のzipとは異なり、最も長いイテラブルの長さに合わせて処理を続行し、要素が不足した部分にはデフォルト値（デフォルトはNone）を補完します。本例では、リストと「スライスで先頭を除いたリスト」を組み合わせることで、各要素とその次の要素をペアとして取得しています。