Сортировать список комплексных чисел и получить индексы сортировки

Question

Мне нужно отсортировать list комплексных чисел по мнимым частям.Я нашел решение по вопросу « сортировка списка комплексных чисел ».

Просто используя отсортированную команду с соответствующей клавишей:

list_ordered = sorted(list, key=lambda x: x.imag) 

Я бы тожехотел бы вернуть индексы сортировки. Другое существующее решение не работает в случае комплексных чисел.Есть ли элегантное решение для извлечения индексов в моем случае?

Спасибо!

Answer 1

Второй ответ на вопрос, который вы связали , можно легко адаптировать:

complexes = [1, 2+3j, 1-2j, 6+1j]
[i[0] for i in sorted(enumerate(complexes), key=lambda x:x[1].imag)]

# [2, 0, 3, 1]