你如何按数字对文件进行排序？

Question

我正在处理目录中的一些文件，需要按数字对文件进行排序。 我在wiki.python.org找到了一些关于排序的例子——特别是使用lambda模式——我把它们放在一起：

#!env/python
import re

tiffFiles = """ayurveda_1.tif
ayurveda_11.tif
ayurveda_13.tif
ayurveda_2.tif
ayurveda_20.tif
ayurveda_22.tif""".split('\n')

numPattern = re.compile('_(\d{1,2})\.', re.IGNORECASE)

tiffFiles.sort(cmp, key=lambda tFile:
                   int(numPattern.search(tFile).group(1)))

print tiffFiles

我对 Python 还是比较陌生，想问社区是否可以对此进行任何改进：缩短代码（删除lambda ）、性能、样式/可读性？

谢谢你，扎卡里

Answer 1

这称为“自然排序”或“人工排序”（与默认的字典排序相反）。 Ned B 写了一个快速版本。

import re

def tryint(s):
    try:
        return int(s)
    except:
        return s

def alphanum_key(s):
    """ Turn a string into a list of string and number chunks.
        "z23a" -> ["z", 23, "a"]
    """
    return [ tryint(c) for c in re.split('([0-9]+)', s) ]

def sort_nicely(l):
    """ Sort the given list in the way that humans expect.
    """
    l.sort(key=alphanum_key)

它类似于您正在做的事情，但可能更笼统一些。

Answer 2

只需使用：

tiffFiles.sort(key=lambda var:[int(x) if x.isdigit() else x for x in re.findall(r'[^0-9]|[0-9]+', var)])

比使用 try/except 更快。

Answer 3

如果您在排序方法中使用key= ，则不应使用已从最新版本的 Python 中删除的cmp 。 key应该等同于一个函数，它将记录作为输入并返回任何对象，该对象将按照您希望列表排序的顺序进行比较。 它不需要是一个 lambda 函数，作为一个独立的函数可能会更清晰。 此外，正则表达式的计算速度可能很慢。

您可以尝试类似以下操作来隔离并返回文件名的整数部分：

def getint(name):
    basename = name.partition('.')
    alpha, num = basename.split('_')
    return int(num)
tiffiles.sort(key=getint)

Answer 4

@April在Pythons glob.glob 的顺序中提供了一个很好的解决方案？ 你可以试试

#First, get the files:
import glob
import re

files = glob.glob1(img_folder,'*'+output_image_format)

# Sort files according to the digits included in the filename
files = sorted(files, key=lambda x:float(re.findall("(\d+)",x)[0]))

Answer 5

元组中的分区结果

def getint(name):
    (basename, part, ext) = name.partition('.')
    (alpha, num) = basename.split('_')
    return int(num)

Answer 6

这是@Don O'Donnell 答案的修改版本，因为我无法让它按原样工作，但我认为这是最好的答案，因为它已经得到了很好的解释。

def getint(name):
    _, num = name.split('_')
    num, _ = num.split('.')
    return int(num)

print(sorted(tiffFiles, key=getint))

变化：

1) alpha字符串不会被存储，因为它不需要（因此_, num ）

2) 使用num.split('.')将数字与 .tiff 分开

3) 使用sorted而不是list.sort ，根据https://docs.python.org/2/howto/sorting.html