Pythonでリスト内の連続する文字列要素をグループ化する方法

リスト内に連続して現れる文字列要素をひとつのグループにまとめたい場合、groupbyとyieldを利用したメソッドを定義することで簡単に実現できます。

サンプルコード

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

from itertools import groupby

def string_check(elem):
    return isinstance(elem, str)

def group_string(my_list):
    for key, grp in groupby(my_list, key=string_check):
        if key:
            yield list(grp)
        else:
            yield from grp

my_list = [52, 11, 'py', 'th', 'on', 11, 52, 'i', 's', 18, 'f', 'un', 99]

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

my_result = [*group_string(my_list)]

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

実行結果

The list is :
[52, 11, 'py', 'th', 'on', 11, 52, 'i', 's', 18, 'f', 'un', 99]
The result is:
[52, 11, ['py', 'th', 'on'], 11, 52, ['i', 's'], 18, ['f', 'un'], 99]

コードの解説

• まず「string_check」という関数を定義します。引数として受け取った要素が文字列型（str）かどうかをisinstanceで判定し、真偽値を返します。

• 次に「group_string」という関数を定義します。この関数はリストを引数に取り、itertoolsモジュールのgroupbyを使って隣接する同種の要素をグループ化し、yieldによって結果を順次返します。文字列のグループはリストとしてまとめられ、それ以外の要素はそのままその場で返されます。

• 関数の外側では、整数と文字列が混在したリストを定義し、コンソールに表示しています。

• 「group_string」を呼び出し、アンパック演算子*を使ってジェネレータの全要素をリスト化し、変数に代入します。

• 最後に、処理結果をコンソールに出力します。

ポイント

itertools.groupbyは「連続している」同じキーを持つ要素だけをまとめるという特徴がある点に注意してください。リスト全体から同種の要素をすべて集めたい場合は、事前にsorted()などで並べ替えてからグループ化する必要があります。