是否可以从列表理解中返回两个列表?好吧,这显然行不通,但是类似于:
rr, tt = [i*10, i*12 for i in xrange(4)]
所以rr
和tt
都是列表,结果分别来自i*10
和i*12
。非常感谢
是否可以从列表理解中返回两个列表?好吧,这显然行不通,但是类似于:
rr, tt = [i*10, i*12 for i in xrange(4)]
所以rr
和tt
都是列表,结果分别来自i*10
和i*12
。非常感谢
>>> rr,tt = zip(*[(i*10, i*12) for i in xrange(4)])
>>> rr
(0, 10, 20, 30)
>>> tt
(0, 12, 24, 36)
创建两个理解列表更好(至少对于长列表)。请注意,最佳投票答案比传统的 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
/generator
usingyield
)。这是一个例子:
$ 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%)。
如果元素是列表,列表推导可能会返回多个列表。例如:
>>> x, y = [[] for x in range(2)]
>>> x
[]
>>> y
[]
>>>
函数的技巧zip
可以完成这项工作,但如果您只是使用循环将结果收集到列表中,实际上会更加简单和易读。