對復數列表進行排序並獲取排序索引

Question

我需要按虛部對復數list進行排序。 我在“ 復數排序列表”問題上找到了解決方案。

只需使用帶有適當鍵的 sorted 命令：

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]