Pythonで選択的な連続サフィックス結合を実現する方法

文字列のリストを扱う際に、特定のサフィックス(接尾辞)を目印として、連続する要素同士を結合したいケースがあります。このような「選択的な連続サフィックス結合」は、単純なforループによる反復処理と、文字列メソッドのendswith、リストメソッドのappendやjoinを組み合わせることで、シンプルに実現できます。

考え方

基本的な仕組みは以下の通りです。

• リストの要素を順番に一時リストへ追加していきます。
• 要素が指定したサフィックスで終わっていない場合、そこがグループの区切りとみなされ、それまでに蓄積した要素をjoinで連結して結果リストに格納します。
• 一時リストを空にして、次のグループの処理に移ります。

サンプルコード

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

my_list = ["Python-", "fun", "to-", "code"]

print("リストの内容 :")
print(my_list)

suffix = '-'
print("サフィックス :")
print(suffix)

result = []
temp = []
for element in my_list:
   temp.append(element)

   if not element.endswith(suffix):
      result.append(''.join(temp))
      temp = []

print("結合結果 :")
print(result)

出力結果

リストの内容 :
['Python-', 'fun', 'to-', 'code']
サフィックス :
-
結合結果 :
['Python-fun', 'to-code']

コードの解説

• まず、結合対象となる文字列のリストを定義し、コンソールに表示します。
• 次に、判定の基準となるサフィックス(ここでは「-」)を定義して表示します。
• 結合結果を格納するresultと、一時的に要素を蓄積するtempという2つの空リストを作成します。
• 元のリストを1要素ずつ反復処理しながら、各要素を一時リストtempに追加していきます。
• 要素が指定したサフィックスで終わっていない場合はグループの終端であるため、joinメソッドでtemp内の要素を連結し、その文字列をresultに追加します。
• 連結後はtempを再度空にすることで、次のグループの蓄積を正しく開始できるようにしています。
• すべての要素を処理し終えたら、最終的な結合結果がコンソールに出力されます。

この例では、「Python-」と「fun」、「to-」と「code」がそれぞれ1つのグループとして扱われ、['Python-fun', 'to-code']という結果が得られます。ハイフンなどの記号を区切り目にして文字列をグループ化したい場合に、この手法は非常に便利です。