[英]pythonic way to associate list elements with their indices
我有一个值列表,我想把它们放在一个字典中,将每个值映射到它的索引。
我可以这样做:
>>> t = (5,6,7)
>>> d = dict(zip(t, range(len(t))))
>>> d
{5: 0, 6: 1, 7: 2}
这不错,但我正在寻找更优雅的东西。
我遇到过以下内容,但它与我需要的相反:
>>> d = dict(enumerate(t))
>>> d
{0: 5, 1: 6, 2: 7}
请分享您的解决方案,
谢谢
编辑 :Python 2.6.4
对于包含1000个元素的列表,dict(zip)版本是最快的,生成器和列表推导版本几乎完全相同,它们慢约1.5倍,功能映射(反向)相当慢。
$ python -mtimeit -s“t = range(int(1e3))”“d = dict(zip(t,range(len(t))))”
1000循环,每循环最佳3:277 usec
$ python -mtimeit -s“t = range(int(1e3))”“d = dict([(y,x)代表x,y代表枚举(t)])”
1000循环,每循环最佳3:426 usec
$ python -mtimeit -s“t = range(int(1e3))”“d = dict(x,(y,x),枚举(t)中的y)”
1000循环,最佳3:437每循环usec
$ python -mtimeit -s“t = range(int(1e3))”“d = dict(map(reverse,enumerate(t)))”
100个循环,最佳3:3.66毫秒/循环
我尝试对较长的列表和较短的列表(1e2,1e4,1e5)运行相同的测试,并且每个循环的时间与列表的长度成线性比例。
有人可以时间py 2.7+版本吗?
您可以使用列表推导(或生成器,具体取决于您的python版本)为您的第二个示例执行简单的就地交换。
使用列表理解:
d = dict([(y,x) for x,y in enumerate(t)])
使用生成器表达式(Python 2.4及更高版本):
d = dict((y,x) for x,y in enumerate(t))
在Python2.7 +中你可以像这样写它
>>> t = (5,6,7)
>>> d = {x:i for i,x in enumerate(t)}
>>> print d
{5: 0, 6: 1, 7: 2}
>>> dict((x,i) for i,x in enumerate(t))
{5: 0, 6: 1, 7: 2}
>>>
您的所有元素都是唯一的(即您的列表永远不会是5,6,7,7)吗? 只有当您的所有元素都是唯一的时,dict解决方案才有效。
通过存储索引,您基本上可以复制信息,因为您可以简单地查询列表中项目的当前索引。 复制信息通常不是最好的主意,因为它允许一组数据与另一组数据不同步。
如果正在修改列表,也没有什么可以防止您意外地将同一索引分配给多个项目。
当您只是从列表中获取索引时,为什么要尝试存储索引值?
正如大家已经写过的那样,在Python 2.6中我会认为以下是最pythonic:
>>> dict((x, i) for i, x in enumerate(t))
{5: 0, 6: 1, 7: 2}
不过,在功能狂热的那一刻,我会写道:
>>> dict(map(reversed, enumerate(t)))
{5: 0, 6: 1, 7: 2}
我最喜欢dict(zip(t,range(len(t))))。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.