Python小技巧:一种字符串的排序方式
1.排序方式
假设有一个序列,数据为:['n1', 'n2', 'n10', 'n11', 'n21', 'n3', 'n13', 'n20', 'n23'],
排序后需要达到这个效果:['n1', 'n2', 'n3', 'n10', 'n11', 'n13', 'n20', 'n21', 'n23']。
最开始当然想的是使用内建函数sorted()进行排序,但效果如下:
l = ['n1', 'n2', 'n10', 'n11', 'n21', 'n3', 'n13', 'n20', 'n23']
for item in sorted(l):
print (item+' ', end='')
运行结果:
n1 n10 n11 n13 n2 n20 n21 n23 n3
很明显,没有达到我们想要的效果,后来上网查资料,发现这篇博客写的可以用来参考,地址:python 根据字符串内数字排序。
实现的思想是通过自定义排序的方式来实现,代码如下,在代码中,使用正则表达式匹配字符串尾部的数字。
import os
import sys
import re
def sort_key(s):
if s:
try:
suffix = re.search('\d+$', s)
num = int(suffix.group())
except:
num = -1
return num
def strsort(alist):
alist.sort(key=sort_key)
return alist
if __name__ == "__main__":
l = ['n1', 'n2', 'n10', 'n11', 'n21', 'n3', 'n13', 'n20', 'n23']
sl = strsort(l)
print (sl)
输出结果:
['n1', 'n2', 'n3', 'n10', 'n11', 'n13', 'n20', 'n21', 'n23']
可以看出,通过这种方法可以实现我们想要的排序方式。
2.扩展
现在,做一些扩展,假设现在需要排序的字符串的格式组成方式为:nxxx_name_nxxx
,也就是由前缀数字和后缀数字组成,例如,我们要把下列的这些字符串进行排序:
n1_name_n1、n2_name_n2、n1_name_n3、n2_name_n1、n3_name_n1、n3_name_n2
,先以前缀数字优先排序,再以后缀数字优先排序。排序后得到的顺序为:
n1_name_n1、n1_name_n3、n2_name_n1、n2_name_n2、n3_name_n1、n3_name_n2
,那么我们可以给前缀数字赋一个很大的权重,再加上后缀,组成的数字就是排序的依据。
实现代码如下,在代码中,我给前缀的数字乘上了255,后缀的数字没有大于255的,所以能够实现上述的排序结果。
import os
import sys
import re
def sort_key(s):
if s:
try:
prefix = re.match('n\d+', s)
suffix = re.search('\d+$', s)
if prefix != None:
prefix_num = int(prefix.group()[1:])
else:
prefix_num = -1
if suffix != None:
suffix_num = int(suffix.group())
else:
suffix = -1
num = prefix_num * 255 + suffix_num
except:
num = -1
return num
def strsort(alist):
alist.sort(key=sort_key)
return alist
#学习中遇到问题没人解答?小编创建了一个Python学习交流群:153708845
if __name__ == "__main__":
l_raw = ["n1_name_n1", "n2_name_n2", "n1_name_n3", "n2_name_n1", "n3_name_n1", "n3_name_n2"]
l_sorted = strsort(l_raw)
print (l_sorted)
输出结果:
['n1_name_n1', 'n1_name_n3', 'n2_name_n1', 'n2_name_n2', 'n3_name_n1', 'n3_name_n2']
这种思想,还可以用来处理多种条件下的排序,还是挺好用的