在列表推导中从函数中解包返回值 [英] Unpacking return value from function in List Comprehension

问题描述

此代码:

def f(v):
    return (v,v+1)

a = [f(i) for i in range(3)]

给予:

[(0, 1), (1, 2), (2, 3)]

我希望更改理解范围，以便提供:

I wish to change the comprehension so that it gives:

[0, 1, 1, 2, 2, 3]

我该怎么做?

推荐答案

在或者，您可以使用 itertools.chain.from_iterable :

Or, you can use itertools.chain.from_iterable:

>>> import itertools
>>> list(itertools.chain.from_iterable(f(i) for i in range(3)))
[0, 1, 1, 2, 2, 3]

这篇关于在列表推导中从函数中解包返回值的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！