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

iBigder

于 2021-07-11 21:22:20 发布

阅读量530

点赞数

CC 4.0 BY-SA版权

分类专栏：日常实用技术笔记文章标签： python

本文链接：https://blog.youkuaiyun.com/unit2006/article/details/118660154

日常实用技术笔记专栏收录该内容

44 篇文章

订阅专栏

本文介绍了一个函数find_genre，用于根据文件扩展名查找并返回相应的IT类型，如文档、图片、音频或压缩包。通过简单的遍历字典，该函数简化了文件类型识别过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

根据 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'))
# 输出：图片