在Python中,我如何自然地对字母数字字符串列表进行排序,使字母字符排在数字字符之前?

时间:2012-08-29 18:11:27

标签: python sorting

这是我最近遇到的一个有趣的小挑战。我将在下面提供我的答案,但我很想知道是否有更优雅或更有效的解决方案。

描述了向我提出的要求:

  1. 字符串是字母数字(参见下面的测试数据集)
  2. 字符串应自然排序(请参阅this question以获取解释)
  3. 字母字符应在数字字符之前排序(即'100'之前的'abc')
  4. alpha字符的大写实例应该在小写实例之前排序(即'ABc','Abc','abc')
  5. 这是一个测试数据集:

    test_cases = [
        # (unsorted list, sorted list)
        (list('bca'), ['a', 'b', 'c']),
        (list('CbA'), ['A', 'b', 'C']),
        (list('r0B9a'), ['a', 'B', 'r', '0', '9']),
        (['a2', '1a', '10a', 'a1', 'a100'], ['a1', 'a2', 'a100', '1a', '10a']),
        (['GAM', 'alp2', 'ALP11', '1', 'alp100', 'alp10', '100', 'alp1', '2'],
            ['alp1', 'alp2', 'alp10', 'ALP11', 'alp100', 'GAM', '1', '2', '100']),
        (list('ra0b9A'), ['A', 'a', 'b', 'r', '0', '9']),
        (['Abc', 'abc', 'ABc'], ['ABc', 'Abc', 'abc']),
    ]
    

    奖金测试用例

    这受到以下Janne Karila's comment的启发,所选答案目前失败(但在我的案例中并不是真正的实际问题):

    (['0A', '00a', 'a', 'A', 'A0', '00A', '0', 'a0', '00', '0a'],
            ['A', 'a', 'A0', 'a0', '0', '00', '0A', '00A', '0a', '00a'])
    

3 个答案:

答案 0 :(得分:6)

re_natural = re.compile('[0-9]+|[^0-9]+')

def natural_key(s):
    return [(1, int(c)) if c.isdigit() else (0, c.lower()) for c in re_natural.findall(s)] + [s]

for case in test_cases:
    print case[1]
    print sorted(case[0], key=natural_key)

['a', 'b', 'c']
['a', 'b', 'c']
['A', 'b', 'C']
['A', 'b', 'C']
['a', 'B', 'r', '0', '9']
['a', 'B', 'r', '0', '9']
['a1', 'a2', 'a100', '1a', '10a']
['a1', 'a2', 'a100', '1a', '10a']
['alp1', 'alp2', 'alp10', 'ALP11', 'alp100', 'GAM', '1', '2', '100']
['alp1', 'alp2', 'alp10', 'ALP11', 'alp100', 'GAM', '1', '2', '100']
['A', 'a', 'b', 'r', '0', '9']
['A', 'a', 'b', 'r', '0', '9']
['ABc', 'Abc', 'abc']
['ABc', 'Abc', 'abc']

编辑:我决定重新审视这个问题,看看是否可以处理奖金案件。它需要在钥匙的断路器部分更复杂。要匹配所需的结果,必须在数字部分之前考虑键的alpha部分。我还在钥匙的自然部分和打破平局之间添加了一个标记,以便短钥匙总是在长钥匙之前。

def natural_key2(s):
    parts = re_natural.findall(s)
    natural = [(1, int(c)) if c.isdigit() else (0, c.lower()) for c in parts]
    ties_alpha = [c for c in parts if not c.isdigit()]
    ties_numeric = [c for c in parts if c.isdigit()]
    return natural + [(-1,)] + ties_alpha + ties_numeric

这会为上面的测试用例生成相同的结果,加上奖金案例的所需输出:

['A', 'a', 'A0', 'a0', '0', '00', '0A', '00A', '0a', '00a']

答案 1 :(得分:2)

这是一个适合奖金测试的人:

def mykey(s):
    lst = re.findall(r'(\d+)|(\D+)', s)
    return [(0,a.lower()) if a else (1,int(n)) for n, a in lst]\
        + [a for n, a in lst if a]\
        + [len(n) for n, a in lst if n]

def mysort(lst):
    return sorted(lst, key=mykey)

使用这种类型的模式,re.findall将字符串分解为元组列表,例如。

>>> re.findall(r'(\d+)|(\D+)', 'ab12cd')
[('', 'ab'), ('12', ''), ('', 'cd')]

答案 2 :(得分:0)

此功能目前不对绩效提出任何要求:

def alpha_before_numeric_natural_sensitive(unsorted_list):
    """presorting the list should work because python stable sorts; see:
    http://wiki.python.org/moin/HowTo/Sorting/#Sort_Stability_and_Complex_Sorts"""
    presorted_list = sorted(unsorted_list)
    return alpha_before_numeric_natural(presorted_list)

def alpha_before_numeric_natural(unsorted_list):
    """splice each string into tuple like so:
    'abc100def' -> ('a', 'b', 'c', 100, 'd', 'e', 'f') ->
    (ord('a'), ord('b'), ord('c'), ord('z') + 1 + 100, ...) then compare
    each tuple"""
    re_p = "([0-9]+|[A-za-z])"
    ordify = lambda s: ord('z') + 1 + int(s) if s.isdigit() else ord(s.lower())
    str_to_ord_tuple = lambda key: [ordify(c) for c in re.split(re_p, key) if c]
    return sorted(unsorted_list, key=str_to_ord_tuple)

它基于this natural sort solution提供的洞察力以及我写的这个函数:

def alpha_before_numeric(unsorted_list):
    ord_shift = lambda c: c.isdigit() and chr(ord('z') + int(c.lower()) + 1) or c.lower()
    adjust_word = lambda word: "".join([ord_shift(c) for c in list(word)])

    def cmp_(a, b):
        return cmp(adjust_word(a), adjust_word(b))

    return sorted(unsorted_list, cmp_)

有关比较不同功能的完整测试脚本,请参阅http://klenwell.com/is/Pastebin20120829