我有一個包含嵌套字典的字典串列，如下所示。我想使用生成器和 yield 來遍歷reviewsdict 串列并回傳所有rating給定的特定user值。例如"user" : "A11"有兩條rating記錄。我想讓這些作為dict回傳。
data = [{ "product_id" : "123",
"size" : "L",
"color" : "blue",
"reviews" : [
{ "user" : "A11", "rating" : "score1" },
{ "user" : "Z99", "rating" : "score" }] },
{ "product_id" : "987",
"size" : "M",
"color" : "red",
"reviews" : [
{ "user" : "A11", "rating" : "score2" },
{ "user" : "X55", "rating" : "score" }] }
]
我有以下生成器，但它基本上回傳所有rating值，而不管user. 如何使函式將值過濾到特定的值user？
def user_rating(nest, kv):
if isinstance(nest, list):
for i in nest:
for x in user_rating(i, kv):
yield x
elif isinstance(nest, dict):
if kv in nest:
yield nest[kv]
for j in nest.values():
for x in user_rating(j, kv):
yield x
print(list(user_rating(data, 'rating')))
uj5u.com熱心網友回復：
稍作修改就足夠了：
def user_rating(nest, kv):
if isinstance(nest, list):
for i in nest:
yield from user_rating(i, kv)
elif isinstance(nest, dict):
if nest.get('user') == kv: # <----- here
yield nest # <---- and here
for j in nest.values():
yield from user_rating(j, kv)
print(list(user_rating(data, 'A11')))
#[{'user': 'A11', 'rating': 'score1'}, {'user': 'A11', 'rating': 'score2'}]
順便說一句，您可以使用yield from...好吧，從另一個生成器產生而不是使用for回圈
轉載請註明出處，本文鏈接：https://www.uj5u.com/qiye/410036.html
標籤：