[英]How to replace all items in a list with their order in python?
让我们看看这个列表:
L = [8,1,4,2]
我想用它们的顺序替换它的元素。
预期结果:
[4,1,3,2]
以下脚本有效,但其双 for 循环确实效率低下:
L_result = []
for i in L :
order = 1
for j in L :
if i > j :
order += 1
L_result.append(order)
L = [8, 1, 4, 2]
positions = {e: i for i, e in enumerate(sorted(L), 1)}
result = [positions[e] for e in L]
print(result)
Output
[4, 1, 3, 2]
这种方法是O(n log n)
,因为正在对数组进行排序。 如果L
有重复值,您可以执行以下操作:
from collections import defaultdict, deque
L = [8, 1, 4, 8, 2]
positions = defaultdict(deque)
for i, e in enumerate(sorted(L), 1):
positions[e].append(i)
result = [positions[e].popleft() for e in L]
print(result)
Output
[4, 1, 3, 5, 2]
使用双端队列的原因是为了使顺序稳定,前 8 个有第一个 position,同时保持 popleft 操作O(1)
,因此算法保持O(n log n)
。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.