python每期一练1019

"""
给定一个键值都是字符串的字典，请编写一个函数，将原来的键和值进行对调。

示例
输入：{'Ice': 'Cream', 'Age': '21', 'Light': 'Cream', 'Double': 'Cream'}，
输出：{'Cream': ['Ice', 'Light', 'Double'], '21': ['Age']}

题目难度：简单
"""


def switch_dict(dc: dict) -> dict:
    result_dict = {}
    for value in dc.values():
        result_dict[value] = []
        for key in dc.keys():
            if dc[key] == value:
                result_dict[value].append(key)
    return result_dict


if __name__ == '__main__':
    data = {'Ice': 'Cream', 'Age': '21', 'Light': 'Cream', 'Double': 'Cream'}
    expected = {'Cream': ['Ice', 'Light', 'Double'], '21': ['Age']}

    assert switch_dict(data) == expected