python list comprehension在一次迭代中产生两个值

我想在python中生成一个列表，如下所示-

[1, 1, 2, 4, 3, 9, 4, 16, 5, 25 .....]

您会发现，这不过是n, n*n

我尝试如下在python中编写这样的列表理解-

lst_gen = [i, i*i for i in range(1, 10)]

但是这样做会产生语法错误。

通过列表理解生成上面的列表的一种好方法是什么？

12个解决方案

44 votes

使用itertools.chain.from_iterable：

>>> from itertools import chain

>>> list(chain.from_iterable((i, i**2) for i in xrange(1, 6)))

[1, 1, 2, 4, 3, 9, 4, 16, 5, 25]

或者，您也可以使用生成器函数：

>>> def solve(n):

... for i in xrange(1,n+1):

... yield i

... yield i**2

>>> list(solve(5))

[1, 1, 2, 4, 3, 9, 4, 16, 5, 25]

Ashwini Chaudhary answered 2020-08-09T21:44:03Z

15 votes

这个问题是古老的，但仅对于好奇的读者，我提出另一种可能性：如第一篇文章所述，您可以轻松地从数字列表中创建一对(i，i ** 2)。 然后，您想弄平这对夫妇。 因此，只需在您的理解中添加扁平化操作即可。

[x for i in range(1, 10) for x in (i,i**2)]

Yann answered 2020-08-09T21:44:23Z

6 votes

一个鲜为人知的技巧：列表推导可以有多个for子句。

例如：

>>> [10*x+y for x in range(4) for y in range(3)]

[0, 1, 2, 10, 11, 12, 20, 21, 22, 30, 31, 32]

在您的特定情况下，您可以执行以下操作：

>>> [x*x if y else x for x in range(5) for y in range(2)]

[0, 0, 1, 1, 2, 4, 3, 9, 4, 16]

Jonathan Mayer answered 2020-08-09T21:44:51Z

5 votes

lst_gen = sum([(i, i*i) for i in range(1, 10)],())

哦，我应该提一下，总和可能会打破一个迭代规则:(

Joran Beasley answered 2020-08-09T21:45:11Z

5 votes

列表理解一次生成一个元素。 您可以选择将循环更改为一次仅生成一个值：

[(i//2)**2 if i % 2 else i//2 for i in range(2, 20)]

或生成元组，然后使用itertools.chain.from_iterable()将列表弄平：

from itertools import chain

list(chain.from_iterable((i, i*i) for i in range(1, 10)))

输出：

>>> [(i//2)**2 if i % 2 else i//2 for i in range(2, 20)]

[1, 1, 2, 4, 3, 9, 4, 16, 5, 25, 6, 36, 7, 49, 8, 64, 9, 81]

>>> list(chain.from_iterable((i, i*i) for i in range(1, 10)))

[1, 1, 2, 4, 3, 9, 4, 16, 5, 25, 6, 36, 7, 49, 8, 64, 9, 81]

Martijn Pieters answered 2020-08-09T21:45:40Z

3 votes

您可以创建一个列表列表，然后使用reduce来加入它们。

print [[n,n*n] for n in range (10)]

[[0, 0], [1, 1], [2, 4], [3, 9], [4, 16], [5, 25], [6, 36], [7, 49], [8, 64], [9, 81]]

print reduce(lambda x1,x2:x1+x2,[[n,n*n] for n in range (10)])

[0, 0, 1, 1, 2, 4, 3, 9, 4, 16, 5, 25, 6, 36, 7, 49, 8, 64, 9, 81]

print reduce(lambda x1,x2:x1+x2,[[n**e for e in range(1,4)]\

for n in range (1,10)])

[1, 1, 1, 2, 4, 8, 3, 9, 27, 4, 16, 64, 5, 25, 125, 6, 36, 216, 7, 49, 343, 8, 64, 512, 9, 81, 729]

Reduce接受一个带有两个参数的可调用表达式，并从前两个项目开始处理一个序列。 然后，最后一个表达式的结果将用作后续调用中的第一项。 在这种情况下，每个列表都一个接一个地添加到列表的第一个列表中，然后作为结果返回该列表。

列表推导使用由“ for var in sequence”表达式定义的变量和序列，使用lambda表达式隐式调用map。 以下是同样的事情。

map(lambda n:[n,n*n],range(1,10))

[[1, 1], [2, 4], [3, 9], [4, 16], [5, 25], [6, 36], [7, 49], [8, 64], [9, 81]]

我不知道减少的更自然的python表达式。

John Hall answered 2020-08-09T21:46:31Z

2 votes

另一种选择，对于某些人来说似乎是不恰当的

>>> from itertools import izip, tee

>>> g = xrange(1, 11)

>>> x, y = tee(g)

>>> y = (i**2 for i in y)

>>> z = izip(x, y)

>>> output = []

>>> for k in z:

... output.extend(k)

...

>>> print output

[1, 1, 2, 4, 3, 9, 4, 16, 5, 25, 6, 36, 7, 49, 8, 64, 9, 81, 10, 100]

iruvar answered 2020-08-09T21:46:51Z

1 votes

另外的选择：

reduce(lambda x,y: x + [y, y*y], range(1,10), [])

Matthew Plourde answered 2020-08-09T21:47:11Z

1 votes

>>> lst_gen = [[i, i*i] for i in range(1, 10)]

>>>

>>> lst_gen

[[1, 1], [2, 4], [3, 9], [4, 16], [5, 25], [6, 36], [7, 49], [8, 64], [9, 81]]

>>>

>>> [num for elem in lst_gen for num in elem]

[1, 1, 2, 4, 3, 9, 4, 16, 5, 25, 6, 36, 7, 49, 8, 64, 9, 81]

这是我的参考[http://docs.python.org/2/tutorial/datastructures.html]

Supriya K answered 2020-08-09T21:47:30Z

1 votes

这个线程有很多技巧。 这是另一个使用一个没有进口的班轮发电机

x = (lamdba : [[(yield i), (yield i**2)] for i in range(10)])()

编辑：这将在Python 3.7中引发DeprecatedWarning并在Python 3.8中引发SyntaxError：[https://docs.python.org/dev/whatsnew/3.7.html#deprecated-python-behavior]

Hielke Walinga answered 2020-08-09T21:47:55Z

0 votes

试试这两个衬垫

lst = [[i, i*i] for i in range(10)]

[lst.extend(i) for i in lst]

根据需要更改数学。

更好

#Change my_range to be the number you want range() function of

start = 1

my_range = 10

lst = [i/2 if i % 2 == 0 else ((i-1)/2)**2 for i in range(start *2, my_range*2 - 1)]

ytpillai answered 2020-08-09T21:48:23Z

0 votes

如前所述，itertools是必经之路。 这是我会做的事情，我发现它更加清楚：

[i if turn else i*i for i,turn in itertools.product(range(1,10), [True, False])]

ntg answered 2020-08-09T21:48:43Z