DjangoBB LoFi version

Полная версия: Разбить список на равные части

Начало » Python для новичков » Разбить список на равные части

izooomrud

Янв. 21, 2021 08:31:59

Как можно разбить список

 a = [1,2,3,4,5,6,7,8,9,10]

что бы в результате получить список вида

 [[1, 2, 3], [4, 5, 6], [7, 8, 9], [10]]

py.user.next

Янв. 21, 2021 09:36:58

Для начала хватит. Дальше можешь оптимизировать, если оно потребуется вообще.

  
>>> def split_list(lst, n):
...     out = []
...     tmp = []
...     for i in lst:
...         tmp.append(i)
...         if len(tmp) == n:
...             out.append(tmp)
...             tmp = []
...     if tmp:
...         out.append(tmp)
...     return out
... 
>>> split_list([1, 2, 3, 4, 5, 6], 1)
[[1], [2], [3], [4], [5], [6]]
>>> split_list([1, 2, 3, 4, 5, 6], 2)
[[1, 2], [3, 4], [5, 6]]
>>> split_list([1, 2, 3, 4, 5, 6], 3)
[[1, 2, 3], [4, 5, 6]]
>>> split_list([1, 2, 3, 4, 5, 6], 4)
[[1, 2, 3, 4], [5, 6]]
>>> split_list([1, 2, 3, 4, 5, 6], 5)
[[1, 2, 3, 4, 5], [6]]
>>> split_list([1, 2, 3, 4, 5, 6], 6)
[[1, 2, 3, 4, 5, 6]]
>>> split_list([1, 2, 3, 4, 5, 6], 7)
[[1, 2, 3, 4, 5, 6]]
>>>

tags: split list

sepomd

Янв. 21, 2021 21:59:14

А зачем нужен второй if?

py.user.next

Янв. 22, 2021 00:29:57

sepomd
А зачем нужен второй if?

Без него в этом примере 5 и 6 пропадут

  
>>> split_list([1, 2, 3, 4, 5, 6], 4)
[[1, 2, 3, 4], [5, 6]]
>>>

Отключил второй if

  
>>> def split_list(lst, n):
...     out = []
...     tmp = []
...     for i in lst:
...         tmp.append(i)
...         if len(tmp) == n:
...             out.append(tmp)
...             tmp = []
...     # if tmp:
...     #     out.append(tmp)
...     return out
... 
>>> split_list([1, 2, 3, 4, 5, 6], 4)
[[1, 2, 3, 4]]
>>>