人生苦短,快学Python!
最近有位读者朋友遇到了一个小问题,私聊找小五答疑。感觉也会有其他同学会遇到,所以干脆分享出来。
PS:目前新书《快学Python》读者群已开启,购书后即可加入获得答疑。
如下图所示,在本地文件夹中有这样一堆视频文件,在这种情况下并不是乱序的。
但是将其上传到网盘中后,就会经常变成乱序。即它们会按照1、10、11、2、20这样排序,并不方便我们按顺序去依次观看。
所以我们希望能够给它们在本地重新命名编号,比如可以按照001、002、003···、020这样排序,就可以避免上面这种尴尬的情况了。
在Python中,os模块可以用来自动处理各种文件和目录,比如复制、移动、重命名和删除等操作。
获取文件列表
在交互式环境中输入如下命令:
import os
path = os.getcwd()
filenames = os.listdir(path)
filenames
输出:
os模块中的getcwd()
函数,使用它可以获取当前工作目录。os模块中的listdir()
函数,可以返回工作目录中的所有文件和子目录。通过这两个函数,我们就获取了当前工作目录下的所有文件。
筛选视频文件
在交互式环境中输入如下命令:
file_mp4s = [i for i in filenames if i.split(".")[-1] == "mp4"]
file_mp4s
输出:
这一步骤,是用来筛选文件列表中的所有mp4文件。用循环+条件太繁琐,而列表生成式则可以用一行语句得到视频文件。
批量重命名
在交互式环境中输入如下命令:
for i in file_mp4s:
new_name = i.split("-")[0].zfill(3) + "-" + i.split("-")[1]
os.rename(i,new_name)
输出:
os模块中的rename()
函数,使用它可以重命名文件。
这里还用到一个字符串函数zfill()
,它会返回指定长度的字符串,原字符串右对齐,前面填充0。所以"1".zfill(3)
的话,会返回'001'
。
精选文章
LIWC vs Python | 文本分析之词典统计法略讲(含代码)