如何使用python编写自定义生成器函数

时间:2016-02-01 17:22:28

标签: python generator

我有这个

for A in [0, -0.25, 0.25, -0.5, 0.5, -0.75, 0.75, -1.0, 1.0, -1.25, 1.25, -1.5, 1.5, -1.75, 1.75, -2.0, 2.0, -2.25, 2.25, -2.5, 2.5, -2.75, 2.75, -3.0, 3.0, -3.25, 3.25, -3.5, 3.5, -3.75, 3.75, -4.0, 4.0, -4.25, 4.25, -4.5, 4.5, -4.75, 4.75, -5.0, 5.0]:

是否可以使用发电机功能?我现在有这个:

def frange(start, stop, step=1.0):
    while start <= stop:
        yield start
        start += step

并使用如下:

for error in self.frange(-2.5, 2.5, 0.25):

但它返回[-2.5, 2.25, ... , 0 , 2.25, 2.5],对于我的程序来说,它很难计算。因为我发现价值几乎为零,但我不知道它可能有多少组合。

我需要从零开始,下一个值必须是减号和正值。 比如[0, -0.25, 0.25...]

4 个答案:

答案 0 :(得分:3)

您可以使用numpy.arange()函数,然后按绝对值对值进行排序:

answer = sorted(np.arange(-5, 5.25, 0.25), key=abs)
print(answer)

<强>输出

[0.0,
 -0.25,
 0.25,
 -0.5,
 0.5,
 ...,
 -4.5,
 4.5,
 -4.75,
 4.75,
 -5.0,
 5.0]

答案 1 :(得分:3)

也许你的意思是生成器而不是lambda:

def opposing_numbers(increment, maximum):
     yield 0
     value = increment
     while value <= maximum:
         yield -value
         yield value
         value += increment

然后将其称为:

opposing_numbers(0.25, 5)

答案 2 :(得分:1)

试试这个:

[(i // 2) * 0.25 * (2*(i&1)- 1) for i in range(1,42)]

答案 3 :(得分:0)

这是一个很酷的衬垫

from itertools import chain

list(chain(*zip([i /4.0 for i in range(10)], [-i/4.0 for i in range(9)])))
  

[0.0,0.0,0.25,-0.25,0.5,-0.5,0.75,-0.75,1.0,-1.0,1.25,-1.25,1.5,-1.5,1.75,-1.75,2.0,-2.0]

相关问题