Пользовательская сортировка списка, фиксируя первый элемент - PullRequest
0 голосов
/ 02 июня 2018

У меня есть список

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

Я хочу отсортировать этот список, исправив первый элемент, т. Е. Если я исправлю 35, отсортированный список должен выглядеть как

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

Если яисправить 158 в качестве первого элемента отсортированный список должен выглядеть следующим образомнайдено, что оно не должно идти перед первым элементом.Есть ли встроенная функция для этого в Python?

Ответы [ 5 ]

0 голосов
/ 02 июня 2018

Комбинированное использование itertools.cycle и itertools.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]
0 голосов
/ 02 июня 2018

Просто определите ключевую функцию, например:

Код:

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]
0 голосов
/ 02 июня 2018

Это сделает работу

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]
0 голосов
/ 02 июня 2018

Быстрое и простое 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])
0 голосов
/ 02 июня 2018

Вы можете отсортировать список и затем восстановить индекс элемента с помощью 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]
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...