У меня есть список
[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 для этого встроенная функция?






Вы можете отсортировать список, а затем восстановить индекс элемента с помощью 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]
Это сделает работу
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]
Быстрое и простое решение 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])
Просто определите ключевую функцию, например:
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]
Совместное использование 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]
Если это что-то вроде быстрой сортировки на месте, существует множество существующих решений.