在python列表理解中分配中间变量

时间:2015-09-03 06:41:40

标签: python dictionary list-comprehension

我有一个字符串列表(字符串和列表都可以很大)。我想仅从列表中的某些元素构建字典。

l = ['aaa 0','bbb 1','ccc 2','ddd 3','abc 1']
wanted = set(['aaa','abc'])
dict([x.split() for x in l if x.split()[0] in wanted])

{'aaa': '0', 'abc': '1'}

这可以在不调用split()两次操作的情况下完成吗?

3 个答案:

答案 0 :(得分:6)

使用嵌套生成器表达式((x.split() for x in l))生成键值对:

>>> l = ['aaa 0', 'bbb 1', 'ccc 2', 'ddd 3', 'abc 1']
>>> wanted = {'aaa', 'abc'}
>>> {key: value for key, value in (x.split() for x in l) if key in wanted}
{'abc': '1', 'aaa': '0'}

答案 1 :(得分:2)

我不知道是否效率更高,但以下语句使用split一次:

{k: n for k, n in dict([x.split() for x in l]).iteritems() if k in wanted}

答案 2 :(得分:1)

您可以将split()操作减少为1。

l = ['aaa 0','bdb 1','ccc 2','ddd 3','abc 1']
dict = { }
s = set(['aaa','abc'])
for elem in l:
    nl = elem.split()
    if nl[0] in s:
        dict[nl[0]] = nl[1]
for elm in dict:
    print elm,dict[elm]