I am using a lambda to modify the behaviour of sort.
sorted(list, key=lambda item:(item.lower(),len(item)))
Sorting a list containing the elements A1,A2,A3,A,B1,B2,B3,B
, the result is A,A1,A2,A3,B,B1,B2,B3
.
My expected sorted list would be A1,A2,A3,A,B1,B2,B3,B
.
I've already tried to include the len(item)
for sorting, which didn't work. How to modify the lambda so that the sort result is instead?
Here is one way to do it:
>>> import functools
>>> def cmp(s, t):
'Alter lexicographic sort order to make longer keys go *before* any of their prefixes'
ls, lt = len(s), len(t)
if ls < lt: s += t[ls:] + 'x'
elif lt < ls: t += s[lt:] + 'x'
if s < t: return -1
if s > t: return 1
return 0
>>> sorted(l, key=functools.cmp_to_key(cmp))
['A1', 'A2', 'A3', 'A', 'B1', 'B2', 'B3', 'B']
Traditionally, lexicographic sort order longer strings after their otherwise identical prefixes (i.e. 'abc' goes before 'abcd').
To meet your sort expectation, we first "fix-up" the shorter string by adding the remaining part of the longer string plus another character to make it the longer of the two:
compare abc to defg --> compare abcgx to defg
compare a to a2 --> compare a2x to a2
The functools.cmp_to_key() tool then converts the comparison function to a key function.
This may seem like a lot of work, but the sort expectations are very much at odds with the built-in lexicographic sorting rules.
FWIW, here's another way of writing it, that might or might not be considered clearer:
def cmp(s, t):
'Alter lexicographic sort order to make longer keys go *before* any of their prefixes'
for p, q in zip(s, t):
if p < q: return -1
if q < p: return 1
if len(s) > len(t): return -1
elif len(t) > len(s): return 1
return 0
The logic is:
My first answer was: just negate the len
criterion to reverse only on that criterion.
sorted(list, key=lambda item:(item.lower(),-len(item))) # doesn't work!
But that doesn't work, because there's a conflict between alpha sort and length. Alpha sort puts small strings first. So length criterion doesn't work.
You need to merge both criteria. There's no clear priority between each other.
I found a way: first compute the max length of your strings, then return the chr(127)
filled (the biggest char provided you're using only ASCII) version of the string as key so smallest strings are filled with big chars in the end: they always come last.
l = ["A","B","A1","A2","A3","B1","B2","B3"]
maxlen = max(len(x) for x in l)
print(sorted(l, key=lambda item:item.lower()+chr(127)*(maxlen-len(item))))
result:
['A1', 'A2', 'A3', 'A', 'B1', 'B2', 'B3', 'B']
BTW don't call your list list
for obvious reasons.
If you love us? You can donate to us via Paypal or buy me a coffee so we can maintain and grow! Thank you!
Donate Us With