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

Question

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

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()操作两次的情况下完成吗？

Answer 1

使用嵌套生成器表达式((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'}