Python按照Windows/Ubuntu文件目录显示顺序读取文件
有时我们想按照Windows/Ubuntu文件显示的顺序读取文件,但是读出来的文件顺序却不一样,例如:
我们想按照顺序读取文件夹名为“0/”下的图片,我们期待读出来的顺序为:
但是实际上读出来的顺序为:
网上有很多解决的办法,包括使用sort对os.listdir后的列表进行排序,但是在文件名相当复杂的情况下,需要对文件名进行切割后再排序,非常复杂。换个角度想,如果我们想和Windows或者Linux的显示顺序一样,我们只需要知道系统使用的什么排序算法即可。
实际上,Windows使用的是Numerical Sorting算法,算法思想为将文件名分割,分为数字字符串与字符串后,再分别比较,若为全数字字符串,则比较其值,否则按字符比较。实现也很方便,在python中也有封装,只需安装natsort:
pip install natsort
然后就可以使用该算法,完整代码如下:
import os
from natsort import os_sorted
file_dir = "0"
fileNames = os.listdir(file_dir)
imgs = os_sorted(fileNames)
for img in imgs:
img_name = file_dir + "/" + img
print(img_name)
结果如下:
经测试,在Ubuntu上也是可行的。