pythonの内包表記で、リストを含むディクショナリのデータを作成する方法

Question

実現したいこと

内包表記で、リストを含むディクショナリのデータを作成する方法について知りたい

発生している問題・分からないこと

内包表記でない下記コード例の出力を出したいです。
内包表記でディクショナリのキーまで重複せずに出るのですが、
そのキーに属するデータをリスト化できませんでした。

該当のソースコード

from dataclasses import dataclass
from collections import defaultdict

@dataclass
class placeSort:
    num : int
    place: str 

pls = []
pls.append(placeSort(num = 3  ,place = 'A')) 
pls.append(placeSort(num = 3  ,place = 'B')) 
pls.append(placeSort(num = 3  ,place = 'A')) 
pls.append(placeSort(num = 2  ,place = 'C')) 
pls.append(placeSort(num = 2  ,place = 'D')) 
pls.append(placeSort(num = 1  ,place = 'E')) 

#＜内包表記でない例＞
res0=defaultdict(list)
lbuf = list()
for item in pls:
    if res0.get(item.num) == None:
        lbuf = list()
        lbuf.append(item.place)
        res0[item.num]=lbuf
    else:
        lbuf =res0.get(item.num)
        lbuf.append(item.place)        
        res0[item.num]=lbuf
print(res0)
#出したい出力
# defaultdict(<class 'list'>, {3: ['A', 'B', 'A'], 2: ['C', 'D'], 1: ['E']}

＜内包表記で、できたところまで＞
#キーの一覧をだす
res1 = [z for z in {x.num for x in pls}]
print(res1)
# [1, 2, 3]

#キーの一覧毎の、リストをまとめたいができません
res2 = [(z,y.place)  for y in pls  for z in {x.num for x in pls} if y.num==z]
print(res2)
#[(3, 'A'), (3, 'B'), (3, 'A'), (2, 'C'), (2, 'D'), (1, 'E')]

試したこと・調べたこと

• teratailやGoogle等で検索した
• ソースコードを自分なりに変更した
• 知人に聞いた
• その他

上記の詳細・結果

res2の出力まで出しましたが、リスト化できませんでした。

補足

特になし

Comments

[quickquip]

XY問題 https://ja.meta.stackoverflow.com/questions/2701/#answer-2702 になっていて、
defaultdictやsetdefaultを使ってfor文で書くのが正着で**内包表記で書くべきではない(書けない)** 処理にしか思えないので「なぜ内包表記で書きたいのか?」が欲しい気がします。ただの気分でないなら、質問者さんに必要な回答を書くためにはその理由の方が大事になるかと

[kazuya_]

ご回答ありがとうございます。
dict型を今まで、使用していましたが、dataclassを代わりに使用して便利にデータを
管理できないかと考えたのが発端です。setdeaultを使用すると簡単にできることを知りませんでした。
ありがとうございます。

Answer 1

内包表記で捨てるリストを生成していることや事前ソートしているのが無駄な処理で

res0 = defaultdict(list)
for item in pls:
    res0[item.num].append(item.place)

か

res0 = {}
for item in pls:
    res0.setdefault(item.num, []).append(item.place)

でよいと思います。

Answer 2

解決済みですが、単純に辞書内包表記を使用した例です。ご参考までに提示します。

from dataclasses import dataclass

@dataclass
class placeSort:
    num : int
    place: str 

pls = []
pls.append(placeSort(num = 3  ,place = 'A')) 
pls.append(placeSort(num = 3  ,place = 'B')) 
pls.append(placeSort(num = 3  ,place = 'A')) 
pls.append(placeSort(num = 2  ,place = 'C')) 
pls.append(placeSort(num = 2  ,place = 'D')) 
pls.append(placeSort(num = 1  ,place = 'E')) 

res = {k:[p.place for p in pls if p.num==k] for k in 
       sorted(set(p.num for p in pls), reverse=True)}
print(res)  # {3: ['A', 'B', 'A'], 2: ['C', 'D'], 1: ['E']}

Answer 3

#出したい出力
# defaultdict(<class 'list'>, {3: ['A', 'B', 'A'], 2: ['C', 'D'], 1: ['E']}

from dataclasses import dataclass
from collections import defaultdict

@dataclass
class placeSort:
    num : int
    place: str 

pls = []
pls.append(placeSort(num = 3  ,place = 'A')) 
pls.append(placeSort(num = 3  ,place = 'B')) 
pls.append(placeSort(num = 3  ,place = 'A')) 
pls.append(placeSort(num = 2  ,place = 'C')) 
pls.append(placeSort(num = 2  ,place = 'D')) 
pls.append(placeSort(num = 1  ,place = 'E')) 

##
res = {}
[res.setdefault(p.num, []).append(p.place) for p in pls]
print(res)

# {3: ['A', 'B', 'A'], 2: ['C', 'D'], 1: ['E']}

追記

itertools.groupby() を使う場合。

from itertools import groupby

keyfunc = lambda p: p.num
res = {k: [p.place for p in g] for k, g in groupby(sorted(pls, key=keyfunc), key=keyfunc)}
print(res)

# {1: ['E'], 2: ['C', 'D'], 3: ['A', 'B', 'A']}

Comments

[kazuya_]

いろいろ有効な方法、ご回答ありがとうございました。
便利ですね。慣れないとわかりにくいですが、活用させていただきます。
今後とも、よろしくお願いいたします。