Python中，os.listdir遍历纯数字文件乱序如何解决

Python中，os.listdir遍历纯数字文件乱序如何解决

日常跑深度学习视觉相关代码时，常常需要对数据集进行处理。许多图像文件名是利用纯数字递增的方式命名。通常所用的排序函数sort()，是按照string进行比较的。例如原始的目录下的文件是：

1.jpg
12.jpg
19.jpg
120.jpg
190.jpg

但经过os.listdir()后的顺序就变为：

1.jpg
12.jpg
120.jpg
19.jpg
190.jpg

解决这个问题，首先看python中sort()函数的具体用法，分别看python2.x和python3.x中的sorted()和sort()函数：

• python2.x
  >>>help(list.sort)

Help on method_descriptor:

sort(...)
L.sort(cmp=None, key=None, reverse=False) -- stable sort IN PLACE;
cmp(x, y) -> -1, 0, 1

>>>help(sorted)

Help on built-in function sorted in module builtin:

sorted(...)
sorted(iterable, cmp=None, key=None, reverse=False) --> new sorted list

• python3.x
  >>>help(list.sort)

Help on method_descriptor:

sort(...)
L.sort(key=None, reverse=False) -> None -- stable sort IN PLACE

>>>help(sorted)

Help on built-in function sorted in module builtins:

sorted(iterable, key=None, reverse=False)
Return a new list containing all items from the iterable in ascending order.

A custom key function can be supplied to customise the sort order, and the
reverse flag can be set to request the result in descending order.

（1）由以上可知python2.x与python3.x的区别仅在于参数cmp，python3.x中取消了对cmp的支持。Python3.x直接忽略cmp这个参数即可，为了保证代码通用性，不建议在今后的编程中使用cmp参数。
（2）sort()和sorted()的区别在于sort函数是list列表中的函数，而sorted可以对list或者iterator进行排序。sorted返回的是一个新的排序对象，而sort则是在原对象上进行排序。

基于以上，可以利用python3.x中的sort(key=None, reverse=False)函数来解决上述问题，首先进行参数说明：

key表示一个带参数的函数；
reverse表示是否反序

另外，python2.x的sorted函数中的两个参数表示：

iterable表示list或iterator;
cmp表示带两个参数的比较函数。

通过lambda expression解决，代码如下：
file = os.listdir(path)
file.sort()
file.sort(key = lambda x: int(x[:-4]))
此处x表示参数，x[:-4]则表示只取文件名进行排序，去掉拓展名，此处拓展名是3个字符。可以根据具体的实际情况进行修改。