繁体   English   中英

pythonic方式将列表元素与其索引相关联

[英]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.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM