Заменить extend на list comprehension

Question

Можно ли заменить этот код:

nums = []
for i in range(int(input())):
    nums.extend(map(int, input().split()))

на list comprehension ?
Предполагал, что можно сделать так:
nums=[*map(int, input().split()) for i in range(int(input()))]
но python говорит:
SyntaxError: iterable unpacking cannot be used in comprehension

Answer 1

попробуйте так:

nums = [i
        for _ in range(int(input()))
        for i in map(int, input().split())]

БЛОГ НА HUSL

Заменить extend на list comprehension

Ответы (1 шт):