Вы можете использовать функцию списков pop()
. Он возвращает один элемент списка и удаляет его из списка.
Когда он удаляется с правой стороны, в вашем случае вы должны специально указать, чтобы взять элемент списка по индексу 0
, вызвав pop(0)
.
Поэтому замена двух строк сверху следующим фрагментом должна сделать все за один шаг:
main_list.append([list_of_letters.pop(0) for _ in range(min(len(list_of_letters), a))])
Обратите внимание, что я прекращаю выталкивать элементы из list_of_letters
, если a
больше, чем остальные предметы в нем, следовательно, min(len(list_of_letters), a)
.