54

是否可以从列表理解中返回两个列表?好吧,这显然行不通,但是类似于:

rr, tt = [i*10, i*12 for i in xrange(4)]

所以rrtt都是列表,结果分别来自i*10i*12。非常感谢

4

3 回答 3

87
>>> rr,tt = zip(*[(i*10, i*12) for i in xrange(4)])
>>> rr
(0, 10, 20, 30)
>>> tt
(0, 12, 24, 36)
于 2012-05-07T09:01:19.457 回答
25

创建两个理解列表更好(至少对于长列表)。请注意,最佳投票答案比传统的 for 循环更慢。列表推导更快更清晰

python -m timeit -n 100 -s 'rr=[];tt = [];' 'for i in range(500000): rr.append(i*10);tt.append(i*12)' 
10 loops, best of 3: 123 msec per loop

> python -m timeit -n 100 'rr,tt = zip(*[(i*10, i*12) for i in range(500000)])' 
10 loops, best of 3: 170 msec per loop

> python -m timeit -n 100 'rr = [i*10 for i in range(500000)]; tt = [i*10 for i in range(500000)]' 
10 loops, best of 3: 68.5 msec per loop

很高兴看到列表推导支持一次创建多个列表。

然而,

如果您可以利用传统循环(准确地说是中间计算),那么您可能会更好地使用循环(或iterator/generatorusingyield)。这是一个例子:

$ python3 -m timeit -n 100 -s 'rr=[];tt=[];' "for i in (range(1000) for x in range(10000)): tmp = list(i); rr.append(min(tmp));tt.append(max(tmp))" 
100 loops, best of 3: 314 msec per loop

$ python3 -m timeit -n 100 "rr=[min(list(i)) for i in (range(1000) for x in range(10000))];tt=[max(list(i)) for i in (range(1000) for x in range(10000))]"
100 loops, best of 3: 413 msec per loop

当然,这些情况下的比较是不公平的;在示例中,代码和计算并不等效,因为在传统循环中存储了一个临时结果(参见tmp变量)。因此,列表推导执行了更多的内部操作(它计算了两次 tmp 变量!但它只慢了 25%)。

于 2018-03-12T10:31:35.760 回答
-2

如果元素是列表,列表推导可能会返回多个列表。例如:

>>> x, y = [[] for x in range(2)]
>>> x
[]
>>> y
[]
>>>

函数的技巧zip可以完成这项工作,但如果您只是使用循环将结果收集到列表中,实际上会更加简单和易读。

于 2012-05-07T09:17:40.950 回答