Python根据扩展名找到文件类型

根据 genre 构造一个函数，通过扩展名在字典中查找所查的类型：

genre = {
    '文档':['.doc', '.docx', '.xls', '.xlsx', '.txt', '.pdf'],
    '图片':['.jpg', '.png', '.gif'],
    '音频':['.mp3', '.aac', '.wav'],
    '压缩包':['.zip', '.rar' ,'.7z']
}

def find_genre(extension):
    for key, value in genre.items():
        if extension in value:
            return key

# 调用函数测试一下
print(find_genre('.jpg'))
# 输出：图片