将列表切分为n个几乎相等长度的部分

Question

我想找一种快速、简洁、符合Python风格的方法，把一个列表分成正好n个差不多大小的部分。

partition([1,2,3,4,5],5)->[[1],[2],[3],[4],[5]]
partition([1,2,3,4,5],2)->[[1,2],[3,4,5]] (or [[1,2,3],[4,5]])
partition([1,2,3,4,5],3)->[[1,2],[3,4],[5]] (there are other ways to slice this one too)

这里有几个答案，讨论了如何遍历列表的切片，链接在这里 Iteration over list slices。这些答案和我想要的很接近，不过它们主要关注的是列表的大小，而我更关心的是列表的数量（其中一些答案还填充了None）。这些答案显然可以简单转换，但我想找一个最佳实践。

同样，有人也在这里指出了一些很好的解决方案，链接在这里 How do you split a list into evenly sized chunks?，解决了一个非常相似的问题，但我更关心的是分区的数量，而不是具体的大小，只要大小差不多就行。再次强调，这些也可以简单转换，但我想找一个最佳实践。

Answer 1

这里有一个和Daniel的版本类似的代码：它尽量把东西分得均匀，但把所有较大的部分放在前面：

def partition(lst, n):
    q, r = divmod(len(lst), n)
    indices = [q*i + min(i, r) for i in xrange(n+1)]
    return [lst[indices[i]:indices[i+1]] for i in xrange(n)]

它还避免使用浮点数运算，因为我总是对此感到不太舒服。 :)

编辑：这里有一个例子，主要是为了和Daniel Stutzbach的解决方案做个对比。

>>> print [len(x) for x in partition(range(105), 10)]
[11, 11, 11, 11, 11, 10, 10, 10, 10, 10]

Answer 2

这只是另一种看法，前提是 [[1,3,5],[2,4]] 这个分组方式在你的例子中是可以接受的。

def partition ( lst, n ):
    return [ lst[i::n] for i in xrange(n) ]

这符合 @Daniel Stutzbach 提到的例子中的要求：

partition(range(105),10)
# [[0, 10, 20, 30, 40, 50, 60, 70, 80, 90, 100],
# [1, 11, 21, 31, 41, 51, 61, 71, 81, 91, 101],
# [2, 12, 22, 32, 42, 52, 62, 72, 82, 92, 102],
# [3, 13, 23, 33, 43, 53, 63, 73, 83, 93, 103],
# [4, 14, 24, 34, 44, 54, 64, 74, 84, 94, 104],
# [5, 15, 25, 35, 45, 55, 65, 75, 85, 95],
# [6, 16, 26, 36, 46, 56, 66, 76, 86, 96],
# [7, 17, 27, 37, 47, 57, 67, 77, 87, 97],
# [8, 18, 28, 38, 48, 58, 68, 78, 88, 98],
# [9, 19, 29, 39, 49, 59, 69, 79, 89, 99]]

Answer 3

def partition(lst, n):
    division = len(lst) / float(n)
    return [ lst[int(round(division * i)): int(round(division * (i + 1)))] for i in xrange(n) ]

>>> partition([1,2,3,4,5],5)
[[1], [2], [3], [4], [5]]
>>> partition([1,2,3,4,5],2)
[[1, 2, 3], [4, 5]]
>>> partition([1,2,3,4,5],3)
[[1, 2], [3, 4], [5]]
>>> partition(range(105), 10)
[[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10], [11, 12, 13, 14, 15, 16, 17, 18, 19, 20], [21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31], [32, 33, 34, 35, 36, 37, 38, 39, 40, 41], [42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52], [53, 54, 55, 56, 57, 58, 59, 60, 61, 62], [63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73], [74, 75, 76, 77, 78, 79, 80, 81, 82, 83], [84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94], [95, 96, 97, 98, 99, 100, 101, 102, 103, 104]]

def partition(lst, n):
    division = len(lst) / n
    return [lst[round(division * i):round(division * (i + 1))] for i in range(n)]

Python 3 版本：