查找嵌套字典中键的总数？

Question

这是一个修改后的练习，取自 http://automatetheboringstuff.com/2e/chapter5/。他们最初的示例要求查找所有客人带来的物品总数，例如每个人带来的“苹果”总数

我想弄清楚如何使该函数返回每个人带来的物品总数，这是我来的代码跟上。 它有效，但该

broughtByPerson += v.get('apples',0)

部分似乎可以简化。如果有 100 个不同的项目，如果不为每个不同的项目都编写上述行，代码会是什么样子？

allGuests = {'Alice': {'apples': 5, 'pretzels': 12},
             'Bob': {'ham sandwiches': 3, 'apples': 2},
             'Carol': {'cups': 3, 'apple pies': 1}}

def totalBrought(guests, names): 
    broughtByPerson = 0             
    for k,v in guests.items():      
        if k == names:              
            broughtByPerson += v.get('apples',0)  
            broughtByPerson += v.get('pretzels',0)
            broughtByPerson += v.get('ham sandwiches',0)
            broughtByPerson += v.get('cups',0)
            broughtByPerson += v.get('apple pies',0)
    return broughtByPerson

print('Alice brought: ' + str(totalBrought(allGuests, 'Alice')))
print('Bob brought: ' + str(totalBrought(allGuests, 'Bob')))
print('Carol brought: ' + str(totalBrought(allGuests, 'Carol')))

Answer 1

您可以将内部字典的值相加：

def totalBrought(guests, name):
    return sum(guests[name].values())

Answer 2

这是一个可能的解决方案：

def totalBrought(guests, name):
    broughtByPerson = 0
    for food, quantity in guests[name].items():
        broughtByPerson += quantity
    return broughtByPerson

Answer 3

您可以使用DataDict。首先安装 ndicts 。

pip install ndicts

然后：

from ndicts.ndicts import DataDict

all_guests = {
    'Alice': {'apples': 5, 'pretzels': 12},
    'Bob': {'ham sandwiches': 3, 'apples': 2},
    'Carol': {'cups': 3, 'apple pies': 1}
}
dd = DataDict(all_guests)

apples = dd.extract["", "apples"].total()

其他解决方案看起来很棒，但这应该更通用，并且可以与任何深度的任何嵌套字典一起使用。