Сортировка списка имен файлов с текстом, номером и подчеркиванием в Python

Question

У меня есть список имен файлов, некоторые из них имеют только текст, некоторые из них имеют текст и номер, а некоторые из них имеют все.

Пример:

[ 'mango_1.jpg', 'dog005.jpg', 'guru_2018_01_01.png', 'dog008.jpg', 'mango_6.jpg', 'guru_2018_5_23.png', 'dog01.png', 'mango_11.jpg', 'mango2.jpg', 'guru_2018_02_5.png', 'guru_2019_08_23.jpg', 'dog9.jpg', 'mango05.jpg' ]

My Код:

import re
## ref: https://blog.codinghorror.com/sorting-for-humans-natural-sort-order/
def sort_nicely( l ):
    """ Sort the given list in the way that humans expect.
    """
    convert = lambda text: int(text) if text.isdigit() else text
    alphanum_key = lambda key: [ convert(c) for c in re.split('([0-9]+)', key) ]
    l.sort( key=alphanum_key )
    return print(l)

Фактический вывод:

['dog01.png', 'dog005.jpg', 'dog008.jpg', 'dog9.jpg', 'guru_2018_01_01.png', 'guru_2018_02_5.png', 'guru_2018_5_23.png', 'guru_2019_08_23.jpg', 'mango2.jpg', 'mango05.jpg', 'mango_1.jpg', 'mango_6.jpg', 'mango_11.jpg']

Ожидаемый результат:

['dog01.png', 'dog005.jpg', 'dog008.jpg', 'dog9.jpg', 'guru_2018_01_01.png', 'guru_2018_02_5.png', 'guru_2018_5_23.png', 'guru_2019_08_23.jpg', 'mango_1.jpg', 'mango2.jpg', 'mango5.jpg', 'mango_6.jpg', 'mango_11.jpg']

Как получить ожидаемый результат?

Answer 1

Насколько я понимаю, вы хотите отсортировать по «тексту», а затем по «дате», которая может существовать в каждом имени файла. Итак, сначала вам нужна функция, которая может разбивать имена файлов на эти два компонента:

def split(n):
    # back-to-front, find first letter index
    for (i, c) in enumerate(reversed(n)):
        if not (c.isdigit() or c == '_'):
            break
    # proper (non-reversed) index
    i = len(n) - i
    # split into name and date
    (n, t) = (n[:i], n[i:])
    # split and remove extra underscores
    t = filter(bool, t.split('_'))
    # convert to integers and return
    return n, tuple(map(int, t))

Затем вам нужна функция, чтобы избавиться от любых нежелательных частей в именах файлов (например, расширений):

import os
def parse(n):
    (n, e) = os.path.splitext(n)
    return split(n)

Теперь вы можете просто использовать это как ключ во встроенной функции sorted:

>>> sorted(l, key = parse)
['dog01.png', 'dog005.jpg', 'dog008.jpg', 'dog9.jpg', 'guru_2018_01_01.png', 'guru_2018_02_5.png', 'guru_2018_5_23.png', 'guru_2019_08_23.jpg', 'mango_1.jpg', 'mango2.jpg', 'mango05.jpg', 'mango_6.jpg', 'mango_11.jpg']

Answer 2

Похоже, что в этом случае вы не придаете никакого значения символу _, измените код так, чтобы исключить это

import re
## ref: https://blog.codinghorror.com/sorting-for-humans-natural-sort-order/
def sort_nicely( l ):
    """ Sort the given list in the way that humans expect.
    """
    convert = lambda text: int(text) if text.isdigit() else text
    alphanum_key = lambda key: [ convert(c.replace("_","")) for c in re.split('([0-9]+)', key) ]
    l.sort( key=alphanum_key )
    return print(l)
    ```