如何编写Python函数计算字典中指定国家的人口列表总和

你可以直接利用Python内置的sum()函数计算列表元素总和，代码如下：

def total_country_population(country_pop_dict, country_name):
    # 从字典中取指定国家对应的人口列表，直接计算总和返回
    return sum(country_pop_dict[country_name])

逻辑说明 #

• 第一步通过country_pop_dict[country_name]拿到对应国家存储的人口数值列表
• 第二步调用sum()方法直接计算列表所有元素的累加和，就是该国家的总人口数

如果需要处理传入的国家名不存在于字典的异常场景，避免程序报错，可以增加判断逻辑：

def total_country_population(country_pop_dict, country_name):
    if country_name in country_pop_dict:
        return sum(country_pop_dict[country_name])
    # 国家不存在时可按需求调整返回值，这里示例返回0
    return 0

使用你给出的测试用例运行上述代码，输出结果完全符合预期。

内容的提问来源于stack exchange，提问作者yowhatsup123