如何根据python中的另一个字典迭代一个字典？

Question

我有一本包含一周饮食信息的字典，我们需要根据输出元数据计算客户食用不同类别食物的天数。输出元数据将食物类别作为键，将饮食中的食物作为值。 1, 0 的值是布尔指标，分别表示客户是否吃饭。

这里的顾客在星期四吃了鸡肉和羊肉，但在求和时，我们只想计算一天的“肉”。元数据中的类别分组将来可能会发生变化，例如如果顾客吃鱼。

你能帮我解决这个问题吗？

diet = {'chicken' : [1, 0, 0, 1, 0, 1, 0],
        'lamb'    : [0, 0, 1, 1, 0, 1, 1],
        'salad'   : [0, 0, 1, 0, 1, 1, 1]}

output_metadata = {'Meat'    : ['chicken', 'lamb'],
                   'Veggies' : ['salad'],
                   'Overall' : ['chicken', 'lamb', 'salad']}

想要的输出是

{'Meat': 5, 'Veggies': 4, 'Overall': 6}

Answer 1

如果您希望在不同的日子检查任何选定类别的非零值，您可以将选定的数组压缩在一起。例如：

def days(diet, categories):
    it = zip(*(diet[c] for c in categories))
    return sum(any(x) for x in it)

现在你可以构建输出字典了：

output = {k: days(Diet, v) for k, v in output_metadata.items()}

作为一个单行者（实际上不要这样做）：

output = {k: sum(any(x) for x in zip(*(Diet[c] for c in v))) for k, v in output_metadata.items()}

但实际上，您拥有的是伪装的数据框。考虑使用 pandas 库来做类似的事情。您将能够单独选择日期和/或类别，并总结您想要的任何统计数据。