通过修复第一个元素自定义列表

时间:2018-06-02 01:54:30

标签: python python-3.x sorting

我有一个清单

[25, 35, 54, 70, 68, 158, 78, 11, 18, 12]

我想通过修复第一个元素对此列表进行排序,即:如果我修复35,则排序列表应该看起来像

[35, 54, 68, 70, 78, 158, 11, 12, 18, 25]

如果我将158作为第一个元素,则排序列表应该看起来像

[158, 11, 12, 18, 25, 35, 54, 68, 70, 78]

基本上我想修复第一个元素,其余的应该按排序顺序排列,如果有一个小于第一个元素的数字,它应该不在第一个元素之前。在Python中是否有可用的内置函数?

5 个答案:

答案 0 :(得分:3)

只需定义一个关键功能,如:

代码:

def sorter(threshold):
    def key_func(item):
        if item >= threshold:
            return 0, item
        return 1, item

    return key_func

这可以通过返回一个元组来实现,使得高于阈值的数字将排在阈值以下的数字之下。

测试代码:

data = [25, 35, 54, 70, 68, 158, 78, 11, 18, 12]
print(sorted(data, key=sorter(70)))

结果:

[70, 78, 158, 11, 12, 18, 25, 35, 54, 68]

答案 1 :(得分:0)

您可以对列表进行排序,然后使用lst.index恢复元素的索引以进行透视。

代码

def pivot_sort(lst, first_element):
    lst = sorted(lst)
    index = lst.index(first_element)

    return lst[index:] + lst[:index]

实施例

lst = [25, 35, 54, 70, 68, 158, 78, 11, 18, 12]

print(pivot_sort(lst , 70))
# prints: [70, 78, 158, 11, 12, 18, 25, 35, 54, 68]

答案 2 :(得分:0)

这将完成工作

a = [25, 35, 54, 70, 68, 158, 78, 11, 18, 12]
a.sort()
index = a.index(35)
a = a[index:] + [:index]

print(a) #[35, 54, 68, 70, 78, 158, 11, 12, 18, 25]

答案 3 :(得分:0)

快速而简单的 numpy 解决方案:

def numpy_roll(arr, elem):
    arr = np.sort(arr)
    return np.roll(arr, len(arr)-np.argwhere(arr==elem)[0])

x
# array([17, 30, 16, 78, 54, 83, 92, 16, 73, 47])

numpy_roll(x, 16)
# array([16, 16, 17, 30, 47, 54, 73, 78, 83, 92])

答案 4 :(得分:0)

itertools.cycleitertools.islice的组合使用。

代码

from itertools import cycle, islice

def pivot_sort(lst, pivot):
    sorted_lst = sorted(lst)
    return list(islice(cycle(sorted_lst), sorted_lst.index(pivot), 2*len(sorted_lst)-lst.index(pivot)))


lst = [25, 35, 54, 70, 68, 158, 78, 11, 18, 12]
pivot = 70

print(pivot_sort(lst, pivot))

# [70, 78, 158, 11, 12, 18, 25, 35, 54, 68]