来回Linspace发生器

来回Linspace发生器

我正在寻找一种生成器函数,该函数将沿一条直线返回点,并给出最小距离k。这很简单,可以使用numpy进行如下操作:

points = np.linspace(start, end, k)


但是,我想将点生成为一种“增加的分辨率”,以便在0到1的行上,生成器将产生:

1/2, 1/4, 3/4, 1/8, 3/8, 5/8, ...


同样,这很容易进行递归操作(只需接受端点并在每半部分调用self),但是我想要一个生成器,它可以实现相同的事情,而不必从头开始就将所有内容都填充到数组中,而无需重复点。

最好的方法是什么?

最佳答案

一种实现此目的的方法是使用:

def infinite_linspace():
    den = 2
    while True:
        for i in range(1,den,2):
            yield i/den
        den <<= 1


因此,这里我们用分子从1迭代到den-1(含),然后将分母加倍。

然后,前15个数字为:

>>> list(islice(infinite_linspace(), 15))
[0.5, 0.25, 0.75, 0.125, 0.375, 0.625, 0.875, 0.0625, 0.1875, 0.3125, 0.4375, 0.5625, 0.6875, 0.8125, 0.9375]
>>> [1/2,1/4,3/4,1/8,3/8,5/8,7/8,1/16,3/16,5/16,7/16,9/16,11/16,13/16,15/16]
[0.5, 0.25, 0.75, 0.125, 0.375, 0.625, 0.875, 0.0625, 0.1875, 0.3125, 0.4375, 0.5625, 0.6875, 0.8125, 0.9375]


我们甚至可以添加更多的智能以相对快速地获得第i个元素:

class Linspace:

    def __iter__(self):
        den = 2
        while True:
            for i in range(1,den,2):
                yield i/den
            den <<= 1

    def __getitem__(self, idx):
        if not isinstance(idx, int):
            raise TypeError('idx should be an integer')
        if idx < 0:
            raise ValueError('idx should be positive')
        den = denn = idx+1
        denn |= den >> 1
        while den != denn:
            den = denn
            denn |= denn >> 1
        denn += 1
        return (2*idx+3-denn)/denn


因此,现在我们可以以对数时间访问第10个,第15个和第123'456个元素:

>>> l = Linspace()
>>> l[9]
0.3125
>>> l[14]
0.9375
>>> l[123455]
0.8837966918945312

关于python - 来回Linspace发生器,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/47998640/

10-10 02:59