重复获取元素在列表中位置的更好方法？
重复获取元素在列表中位置的更好方法？
提问于 2012-11-09 13:06:55
我有一个整数列表(从0到N)，我需要创建一个新列表，其中包含第一个列表中每个整数的索引。
也就是说，给定
s = [4, 2, 6, 3, 0, 5, 1]确定r，使s[r[i]] = i
r = [4, 6, 1, 3, 0, 5, 2]我目前的解决方案是
r = [s.index(i) for i in xrange(len(s))]有没有更好的方法？
回答 5
Stack Overflow用户
回答已采纳
发布于 2012-11-09 13:26:10
我假设S中的每个整数恰好出现一次。您当前的解决方案是有效的，问题是s.index执行O(N)搜索，这使得这是一个O(N**2)操作。
对于大型列表，我希望下面的代码会更快，因为它是O(N)
# initialise the whole list with some value
r = [-1]*N
for j, s_j in enumerate(s):
r[s_j] = j
# if any element of r is still -1 then you know it did not appear in sStack Overflow用户
发布于 2012-11-09 13:20:54
对于这一点，字典似乎更好：
s = [4, 2, 6, 3, 0, 5, 1]
r = dict((v,i) for i,v in enumerate(s))测试：
>>> for i,_ in enumerate(s):
... print i, s[r[i]]
...
0 0
1 1
2 2
3 3
4 4
5 5
6 6Stack Overflow用户
发布于 2012-11-09 13:22:26
就我个人而言，你展示的方法很棒。
两个字典都可以--这是我的第一次尝试：
r = {v:i for i, v in enumerate(s)}或者，如果您必须使用列表，另一种方法是：
r = [x[0] for x in sorted(enumerate(s), key=lambda v:v[1])]页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/13302390
复制相关文章
相似问题
腾讯云开发者
