在Python中编写for循环的最优雅方法是什么?

时间:2012-01-02 03:26:40

标签: python math loops for-loop

基本上我想在n步骤中从-1到1,包括-1和1:

x = -1.0
n = 21

for i in range(n):
    print x
    x += 0.01

-1.0 -0.9 -0.8 ... 0.8 0.9 1.0

如何以最优雅,最简单的方式为任何n值编写此内容?

5 个答案:

答案 0 :(得分:11)

没有内置解决方案,但解决它的一个好方法是定义自己的range函数:

def my_range(start, end, how_many):
    incr = float(end - start)/(how_many - 1)
    return [start + i*incr for i in range(how_many)]

你可以在for循环中使用它:

>>> for i in my_range(-1, 1, 10):
...   print i
... 
-1.0
-0.777777777778
-0.555555555556
-0.333333333333
-0.111111111111
0.111111111111
0.333333333333
0.555555555556
0.777777777778
1

编辑:正如@NiklasBaumstark建议的那样,如果您的全新my_range函数要处理大量数字,那么使用generators可能是个好主意。为此,我们只做一点修改:

def my_xrange(start, end, how_many):
        incr = float(end - start)/(how_many - 1)
        return (start + i*incr for i in xrange(how_many))

答案 1 :(得分:11)

如果可以使用numpy,这样可以正常工作:

import numpy as np
n = 21

for i in np.linspace(-1, 1, n):
    print i

答案 2 :(得分:5)

startValue = -1.0
intervalLength = 2.0
numberOfSteps = 14
increasePerStep = intervalLength / numberOfSteps

print startValue

x = startValue
for i in range (numberOfSteps):
    x += increasePerStep
    print x


-1.0
-0.857142857143
-0.714285714286
-0.571428571429
-0.428571428571
-0.285714285714
-0.142857142857
-2.22044604925e-16
0.142857142857
0.285714285714
0.428571428571
0.571428571429
0.714285714286
0.857142857143
1.0

答案 3 :(得分:3)

我会选择

for x in xrange(n):
    print float(2*x)/(n-1) - 1

感兴趣的两件事:xrangerange更有效,并且不需要有两个独立的迭代器变量。

如果你发现它经常有用,你可以将它包装在一个函数中:

def linspace1(n):
    for x in xrange(n):
        yield float(2*x)/(n-1) - 1

虽然您可能也想制作下限和上限参数,例如julio.alegria的回答。

答案 4 :(得分:2)

你说你有答案,但我觉得这是一个优雅的解决方案。它还消除了奇怪浮点问题的可能性(至少从定义的范围)。它结合了生成器的强大功能和灵活性,以及​​字符串格式和浮点值。或者,如果您选择使用Decimal模块,则可以避免大部分内容 - 但这需要更多调整。

def decimal_stepper(start, end, step=0.1):
    while start <= end:
        yield float(start)
        start = float(('%' + str(step) + 'f') % (start+step))

示例运行:

myVals = decimal_stepper(-1, 1)
for x in myVals:
    print x

哪个输出:

-1.0
-0.9
-0.8
-0.7
-0.6
-0.5
-0.4
-0.3
-0.2
-0.1
0.0
0.1
0.2
0.3
0.4
0.5
0.6
0.7 
0.8
0.9
1.0