python文件自动分类(3)

接上节课内容

但在前面判断文件类型的代码中,我们只使用了小写的文件后缀名去进行匹配。
这样的话,比如像“心动.MP3”这个文件就被分类到了“其他文件”。
那么该怎样对大写的后缀名 .MP3 等进行判断呢?

统一字符串

为了解决这个问题,我们可以对文件后缀名使用 lower() 函数。
lower() 函数可以把字符串中所有的字母全部转换成小写。
本示例中,就将两个示例字符串转换成了小写并使用print输出结果。

sampleStr1 = "NATE".lower()

sampleStr2 = "Drake".lower()

print(sampleStr1)

print(sampleStr2)

对文件后缀名使用lower()函数,修改对文件分类的代码,使“心动.MP3”文件被正确分类。

学习了 lower() 函数后,我们现在把它应用到获取文件后缀名的代码中,将文件后缀名全部转换为小写
转换之后,再使用之前的代码进行判断,“心动.MP3”就能被正确分类了。

# 使用import导入os模块

import os

# 将阿文的下载文件夹路径 /Users/yequ/Downloads 赋值给变量downloadPath

downloadPath = "/Users/yequ/Downloads"

# 使用os.listdir()函数获取该路径下所有的文件(夹),并赋值给变量allItems

allItems = os.listdir(downloadPath)

# 使用for循环遍历所有文件(夹)

for item in allItems:

    # 获取文件后缀名并使用lower()函数转换成小写

    extension = os.path.splitext(item)[1].lower()

    if extension in [".jpg", ".jpeg", ".gif", ".png", ".bmp"]:

        print(f"{item} 图片文件")

    elif extension in [".avi", ".mp4", ".wmv", ".mov", ".flv"]:

        print(f"{item} 视频文件")

    elif extension in [".wav", ".mp3", ".mid", ".ape", ".flac"]:

        print(f"{item} 音频文件")

    elif extension in [".pdf"]:

        print(f"{item} PDF文件")

    elif extension in [".docx", ".doc"]:

        print(f"{item} Word文件")

    elif extension in [".xlsx", ".xls"]:

        print(f"{item} Excel文件")

    elif extension in [".pptx", ".ppt"]:

        print(f"{item} PPT文件")

    else:

        print(f"{item} 其他文件")

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值