python - 使用列表理解从 python 函数输出多个列表

Question

这个问题已经有答案了:

Transpose/Unzip Function (inverse of zip)? (14 个回答)

已关闭 3 年前。

我想构建一个函数，为我提供两个列表作为输出:

def times_10(i):
    out = i * 10
    return out, i
​
outpt,inpt = [times_10(i) for i in range(1,8)] #doesn't work

我有一种方法可以做到这一点，但我有一种感觉，一定有更简单的方法:

a = [times_10(i) for i in range(1,8)]
outpt,inpt = list(map(list, zip(*a))) # this works

预期输出:

>>> outpt
[10, 20, 30, 40, 50, 60, 70]
>>> inpt
[1, 2, 3, 4, 5, 6, 7]

Answer 1

您可以解压推导式:

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

示例:

>>> out, inp = zip(*[(i*10, i) for i in range(10)])
>>> out
(0, 10, 20, 30, 40, 50, 60, 70, 80, 90)
>>> inp
(0, 1, 2, 3, 4, 5, 6, 7, 8, 9)