繁体   English   中英

Python列表理解,解包和多个操作

[英]Python list comprehension, unpacking and multiple operations

我想通过执行以下操作来解压缩我创建的元组,因此结果只是一个简单的列表。 我可以在2-3行中获得所需的结果但肯定有一个oneliner list.comp?

x = range(10)
y = [(i,j**2) for i,j in zip(x,x)]
>>>y
[(0, 0), (1, 1), (2, 4), (3, 9), (4, 16), (5, 25), (6, 36), (7, 49), (8, 64), (9
, 81)]
>>>

我想要的是result = [0,0,1,1,2,4,3,9.....]

y = len(x)*[0]
y[::2] = x
y[1::2] = [i**2 for i in x]

给出我想要的但是如果我需要更一般的情况怎么办:

y = [(i, sqrt(i), i**3, some_operation_on_i, f(i), g(i)) for i in x]

例如,我应该能够得到一个像结果一样的直接列表,其中我只指定了一个操作(正方形)来跟随每个i但现在每个i之后有任意数量的操作。

使用嵌套列表理解:

result = [a for tup in y for a in tup]

例:

>>> x = range(10)
>>> y = [(i,j**2) for i,j in zip(x,x)]
>>> [a for tup in y for a in tup]
[0, 0, 1, 1, 2, 4, 3, 9, 4, 16, 5, 25, 6, 36, 7, 49, 8, 64, 9, 81]

这对于更一般的情况也适用,或者您可以一步完成所有操作:

y = [a for i in x for a in (i, sqrt(i), i**3, some_operation_on_i, f(i), g(i))]

如果嵌套列表内涵看起来很奇怪,这里是如何做到这一点看起来像一个正常for循环:

y = []
for i in x:
    for a in (i, sqrt(i), i**3, some_operation_on_i, f(i), g(i)):
        y.append(a)
>>> import itertools
>>> list(itertools.chain.from_iterable(y))
[0, 0, 1, 1, 2, 4, 3, 9, 4, 16, 5, 25, 6, 36, 7, 49, 8, 64, 9, 81]

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

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