假设有两个数组,其中的元素都是数字,现在要对其排序

泄露秘密 提交于 2020-02-26 18:51:21

假设有 number,group,两个数组,其中的元素都是数字,现在要对其排序排序的规则如下:

1.如果 group里面元素,存在于numbers了里面

2.要把出现在 group 内的数字 放在 number 的那些数字之前,注意排序

numbers = [8, 3 ,1, 2, 5,4,7,6]

group = {2,3,5,7}

分析了下,应该是优先级的问题,这里使用了元组排序的方法:

numbers = [8, 3 ,1, 2, 5,4,7,6]
group = {2,3,5,7}
tmp = []
for x in numbers:
    if x in group:
        tmp.append((0, x))
    else:
        tmp.append((1, x))
tmp.sort()
print(list(map(lambda x:x[1], tmp)))

在网上看到了一个经典的代码:

def sort_priority(values,group):
    def helper(x):
        if x in group:
            return (0, x)
        return (1, x)
    values.sort(key=helper)

numbers = [8, 3 ,1, 2, 5,4,7,6]

group = {2,3,5,7}
sort_priority(numbers, group)
print(numbers)

推荐使用第二个高阶函数的方法

标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!