Понимание нескольких списков Python

Question

def test():
    return 1,2

a, b = [test() for _ in range(5)]

Как мне получить a=[1,1,1,1,1] и b=[2,2,2,2,2], используя понимание списка?

Answer 1

zip() является ключом к этому как:

Код:

a, b = [list(x) for x in zip(*[test() for _ in range(5)])]

Код теста:

def test():
    return 1, 2

a, b = [list(x) for x in zip(*[test() for _ in range(5)])]
print(a, b)

Результаты:

[1, 1, 1, 1, 1] [2, 2, 2, 2, 2]