Question

问题的具体实例
我的int范围是1-100。我希望在此范围内生成n个总数尽可能均匀分布并包含第一个和最后一个值。

示例

start = 1, end = 100, n = 5   
Output: [1, 25, 50, 75, 100]

start = 1, end = 100, n = 4   
Output: [1, 33, 66, 100]

start = 1, end = 100, n = 2   
Output: [1, 100]

我目前拥有什么
我实际上有一个工作方法，但我一直觉得我在思考这个并且错过了一些更简单的东西？这是最有效的方法还是可以改进？

def steps(start, end, n):
    n = min(end, max(n, 2) - 1)
    mult = end / float(n)
    yield start
    for scale in xrange(1, n+1):
        val = int(mult * scale)
        if val != start:
            yield val

注意，我确保此函数始终至少返回范围的下限和上限值。所以，我强制n >= 2

仅供搜索引用，我使用它来渲染渲染序列中的图像帧，您通常需要第一个，中间的，最后一个。但我希望能够更好地扩展以处理真正长的图像序列并获得更好的覆盖率。

已解决：来自所选答案

我最终使用这个稍微修改过的@ vartec的答案版本，作为一个生成器，并为安全限制了n值：

def steps(start,end,n):
    n = min(end, max(n, 2))
    step = (end-start)/float(n-1)
    return (int(round(start+x*step)) for x in xrange(n))

Answer 1

你需要适当的舍入：

def steps(start,end,n):
    if n<2:
        raise Exception("behaviour not defined for n<2")
    step = (end-start)/float(n-1)
    return [int(round(start+x*step)) for x in range(n)]

Answer 2

额外的依赖性，可能是矫枉过正，但是很短，经过测试，应该给出正确的结果：numpy.linspace

>>> numpy.linspace(1, 100, 4).astype(int).tolist()
[1, 34, 67, 100]

Answer 3

使用range的问题是该步骤必须是整数，因此您会遇到舍入问题，例如steps(1,100,4) == [1, 33, 66, 100]。如果你想要整数输出但想要尽可能的步骤，可以使用float作为步骤。

>>> def steps(start,end,n):
...    step = (end-start)/float(n-1)
...    return [int(round(start+i*step)) for i in range(n)]

>>> steps(1,100,5)
>>> [1, 26, 51, 75, 100]
>>> steps(1,100,4)
>>> [1, 34, 67, 100]
>>> steps(1,100,2)
>>> [1, 100]
>>>

Answer 4

>>> from itertools import count
>>> def steps(start,end,n):
        yield start
        begin = start if start>1 else 0
        c = count(begin,(end-begin)/(n-1))
        next(c)
        for _ in range(n-2):
            yield next(c)
        yield end


>>> list(steps(1,100,2))
[1, 100]
>>> list(steps(1,100,5))
[1, 25, 50, 75, 100]
>>> list(steps(1,100,4))
[1, 33, 66, 100]
>>> list(steps(50,100,3))
[50, 75, 100]
>>> list(steps(10,100,10))
[10, 20, 30, 40, 50, 60, 70, 80, 90, 100]

可以缩短为

>>> from itertools import islice, count
>>> def steps(start,end,n):
        yield start
        begin = start if start>1 else 0
        c = islice(count(begin,(end-begin)/(n-1)),1,None)
        for _ in range(n-2):
            yield next(c)
        yield end

Answer 5

使用range有什么问题？以下是如何使用它

>>> def steps(start,end,n):
    return [start]+range(start-1,end,end/(n-1))[1:]+[end]

>>> steps(1,100,5)
[1, 25, 50, 75, 100]
>>> steps(1,100,2)
[1, 100]
>>>

在一定范围内生成均匀分布的倍数/样本

5 个答案: