Однострочник возвращает список списков, учитывая список и размер куска:
>>> lol = lambda lst, sz: [lst[i:i+sz] for i in range(0, len(lst), sz)]
Тестирование:
>>> x = range(20, 36)
>>> print x
[20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35]
>>> lol(x, 4)
[[20, 21, 22, 23],
[24, 25, 26, 27],
[28, 29, 30, 31],
[32, 33, 34, 35]]
>>> lol(x, 7)
[[20, 21, 22, 23, 24, 25, 26],
[27, 28, 29, 30, 31, 32, 33],
[34, 35]]
Обновление:
* 1009Я думаю, что вопрос, который действительно задают, это функция, которая, учитывая список и число, возвращает список, содержащий списки $ (число), с равномерно распределенными элементами исходного списка.Таким образом, ваш пример lol (x, 7) должен действительно возвращать [[20,21,22], [23,24,25], [26,27], [28,29], [30,31], [32], 33], [34,35]].- markrian
В данном случае вы можете попробовать:
def slice_list(input, size):
input_size = len(input)
slice_size = input_size / size
remain = input_size % size
result = []
iterator = iter(input)
for i in range(size):
result.append([])
for j in range(slice_size):
result[i].append(iterator.next())
if remain:
result[i].append(iterator.next())
remain -= 1
return result
Я уверен, что это можно улучшить, но мне лень.: -)
>>> slice_list(x, 7)
[[20, 21, 22], [23, 24, 25],
[26, 27], [28, 29],
[30, 31], [32, 33],
[34, 35]]