Pythonでのバイナリ要素リストのグループ化

各サブリストに2つの要素があるリストのリストがあるとします。各サブリストの1つの要素は、リストの他の多くのサブジェクトに共通です。共通の要素でグループ化されたサブリストを表示する最終的なリストを作成する必要があります。

セットとマップ付き

指定されたリストでは、最初の要素は文字列で、2番目の要素は数値です。したがって、各サブリストの2番目の要素を保持する一時リストを作成します。次に、サブリストを一時リストの各要素と比較し、デザイナーがそれらをグループ化します。

例

listA = [['Mon', 2], ['Tue', 3], ['Wed', 3],
["Thu", 1], ['Fri', 2], ['Sat', 3],
['Sun', 1]]
# With set and map
temp = set(map(lambda i: i[1], listA))
res = [[j[0] for j in listA if j[1] == i] for i in temp]
# Result
print("The list with grouped elements is : \n" ,res)

出力

上記のコードを実行すると、次の結果が得られます-

The list with grouped elements is :
[['Thu', 'Sun'], ['Mon', 'Fri'], ['Tue', 'Wed', 'Sat']]

groupbyとitemgetterを使用

itemgetter関数は、各サブリストの2番目の要素を取得するために使用されます。次に、itemgetter関数の結果に基づいてgroupby関数が適用されます。

例

from itertools import groupby
from operator import itemgetter
listA = [['Mon', 2], ['Tue', 3], ['Wed', 3],["Thu", 1], ['Fri', 2], ['Sat', 3],['Sun', 1]]
# With groupby
listA.sort(key = itemgetter(1))
groups = groupby(listA, itemgetter(1))
res = [[i[0] for i in val] for (key, val) in groups]
# Result
print("The list with grouped elements is : \n" ,res)

出力

上記のコードを実行すると、次の結果が得られます-

The list with grouped elements is :
[['Thu', 'Sun'], ['Mon', 'Fri'], ['Tue', 'Wed', 'Sat']]

defaultdictを使用

defaultdict関数を適用して、サブリストの2番目の要素を辞書のキーとして取得します。次に、結果のリストに、サブリストの最初の要素の値を追加します。

例

import collections
listA = [['Mon', 2], ['Tue', 3], ['Wed', 3],["Thu", 1], ['Fri', 2], ['Sat', 3],['Sun', 1]]
# With defaultdict
res = collections.defaultdict(list)
for val in listA:
   res[val[1]].append(val[0])
# Result
print("The list with grouped elements is : \n" ,res)

出力

上記のコードを実行すると、次の結果が得られます-

The list with grouped elements is :
defaultdict(<class 'list'>, {2: ['Mon', 'Fri'], 3: ['Tue', 'Wed', 'Sat'], 1: ['Thu', 'Sun']})