有没有更好的方法在循环内遍历字典中的键?

时间:2021-02-10 21:48:23

标签: python dictionary key

我有一个脚本来计算食谱的营养信息。用户输入配料名称、配料比例(以克为单位),并提供一个 .txt 文件,其中存储了每种配料的营养信息(每 100g),它实际上是一个字典,其键是配料和不同的值(kcal、kj、fat 等)是每个键内的列表。 举个例子,假设成分是茄子、橄榄油和柠檬汁。

到目前为止,脚本将具有以下信息:

nutrition_dict = {'kcal': 0, 'kj': 0, 'fat': 0, 'saturated fat': 0, 'carbohydrates': 0, 'sugar': 0, 'protein': 0, 'salt': 0}
nutrition_file = {'eggplant':[24, 100, 0.2, 0, 5.7, 2.4, 1, 0.005], 'olive oil':[884, 3701, 100, 13.8, 0, 0, 0, 0.005], 'lemon juice':[25, 105, 0, 0, 8.6, 2.4, 0.4, 0.0025]}
amount_of_ingredients = {'eggplant': 300, 'olive oil': 20, 'lemon juice': 5}
total_desired_recipe = 325

现在我需要做的是:

  • 计算总配方的总大卡、kj、脂肪等(本例中为 325 克)
  • 计算每 100 克食谱的总大卡、kj、脂肪等。

我的脚本有效,但它非常丑陋,改进它会让我学习更好的方法来获得所需的结果。

for key in nutrition_file:
    i = 0
    for i in range(8):
        if i == 0:
            nutrition_dict['kcal'] = nutrition_dict['kcal'] + nutrition_file[key][i]
        elif i == 1:
            nutrition_dict['kj'] = nutrition_dict['kj'] + nutrition_file[key][i]
        elif i == 2:
            nutrition_dict['fat'] = nutrition_dict['fat'] + nutrition_file[key][i]
        elif i == 3:
            nutrition_dict['saturated fat'] = nutrition_dict['saturated fat'] + nutrition_file[key][i]
        elif i == 4:
            nutrition_dict['carbohydrates'] = nutrition_dict['carbohydrates'] + nutrition_file[key][i]
        elif i == 5:
            nutrition_dict['sugar'] = nutrition_dict['sugar'] + nutrition_file[key][i]
        elif i == 6:
            nutrition_dict['protein'] = nutrition_dict['protein'] + nutrition_file[key][i]
        elif i == 7:
            nutrition_dict['salt'] = nutrition_dict['salt'] + nutrition_file[key][i]
        i += 1

    for key, value in nutrition_dict.items():
        print("The total of {} is: {:.2f}".format(key, value))
        nutrition = (value * 100) / total_desired_recipe
        print("The amount of {} per 100g is: {:.2f}".format(key, nutrition))
        i += 1

所以我的问题是:有没有更好的方法来遍历 Nutrition_dict 键?

我还希望打印语句是“总信息”并遍历所有内容,然后是“每 100g 信息”并遍历所有内容。我不喜欢目前的“总计,每 100 克,总计,每 100 克”

3 个答案:

答案 0 :(得分:2)

有,您可能只想zip nutrition_dict 键与 nutrition_file 中每种食物的值:

for k, *v in zip(nutrition_dict, *nutrition_file.values()):
    print(k, v)

kcal [24, 884, 25]
kj [100, 3701, 105]
fat [0.2, 100, 0]
saturated fat [0, 13.8, 0]
carbohydrates [5.7, 0, 8.6]
sugar [2.4, 0, 2.4]
protein [1, 0, 0.4]
salt [0.005, 0.005, 0.0025]

然后你需要做的就是收集总数:

for k, *v in zip(nutrition_dict, *nutrition_file.values()):
    nutrition_dict[k] = sum(v)


nutrition_dict
{'kcal': 933, 'kj': 3906, 'fat': 100.2, 'saturated fat': 13.8, 'carbohydrates': 14.3, 'sugar': 4.8, 'protein': 1.4, 'salt': 0.0125}

答案 1 :(得分:0)

您正在寻找的是枚举,它允许您(在本例中)浏览字典的键,同时按顺序对它们进行编号。完整的解决方案是

for index, key in enumerate(nutrition_dict):
    for ingredient_name, ingredient_amount in amount_of_ingredients.items():
        nutrition_dict[key] += ingredient_amount * nutrition_file[ingredient_name][index]

    print("The total of {} is: {:.2f}".format(key, nutrition_dict[key]))
    nutrition = (nutrition_dict[key] * 100) / total_desired_recipe
    print("The amount of {} per 100g is: {:.2f}".format(key, nutrition))

顺便说一句,kJ 和 kcal 在不同单位中是相同的量(因子为 4.184),因此无需跟踪两者。

答案 2 :(得分:0)

在迭代方面,Python 有很多很棒的方法来简化列表处理。事实上,您的大部分循环代码都可以通过消除其他编程语言的一些典型“开销”来实现:

for food_type in nutrition_file:
    for index, metric in enumerate(nutrition_dict):
        nutrition_dict[metric] += nutrition_file[food_type][index]

    # Unchanged from OP's example
    for key, value in nutrition_dict.items():
        print("The total of {} is: {:.2f}".format(key, value))
        nutrition = (value * 100) / total_desired_recipe
        print("The amount of {} per 100g is: {:.2f}".format(key, nutrition))
相关问题