Сделайте один код лайнера для распаковки списка в Python

Question

max_depth = [[i]*36 for i in range(1,11)]
max_depth = [j for k in max_depth for j in k]

Я хочу преобразовать вышеуказанный код в одну строку.

Answer 1

Вместо создания вложенного списка с [i]*36, добавьте еще for для извлечения i 36 раз:

[i for i in range(1,11) for _ in range(36)]

---

Что будет эквивалентно:

max_depth = []
for i in range(11):
    for _ in range(36):
        max_depth.append(i)

Answer 2

Использование модуля itertools:

from itertools import chain

max_depth = list(chain.from_iterable([i]*36 for i in range(1,11)))

Переход на один шаг вперед с помощью itertools,

from itertools import chain, repeat

max_depth = list(chain.from_iterable(repeat(i, 36) for i in range(1,11)))

Answer 3

Самое простое решение:

[j for k in [[i]*36 for i in range(1,11)] for j in k]