在 Python 中将范围分成 n 个相等范围的最佳方法

Question

我有范围内的元素总数N和一些块nb

我想N分成nb最好的相等范围，只有开始编号和结束编号。例如，N=24并且nb=5应该输出：

0,5 5,10 10,15 15,20 20,24

WhileN=28和nb=5应输出：

0,5 5,10 10,16 16,22 22,28  (the rest of `N/nb` division is equally distributed on the 3 last subranges)

基于一个评论，我有这个方法：

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

def ranges(N, nb):
    return ["{},{}".format(r.start, r.stop) for r in partition(range(N), nb)]

>>> ranges(28, 5)
['0,6', '6,11', '11,17', '17,22', '22,28']

有一个更好的方法吗？

Answer 1

直接计算开始和停止数字肯定比切片range对象来获得它们更简单：

def ranges(N, nb):
    step = N / nb
    return ["{},{}".format(round(step*i), round(step*(i+1))) for i in range(nb)]

这并不比您的代码看起来更高效，因为切片range对象只需要O(1)时间，因此您现有的代码已经是渐近最优的。我的版本可能会通过一些常数因素提高性能，但它可能很小。我确实认为我的版本也更清晰，这可能比可能出现的任何性能变化更重要。