0

我正在尝试在 python3 中优化这段代码以提高速度(当然还有正确性):

from math import log
from timeit import Timer

def groffle_slow(mass, density):
    total = 0.0
    for i in range(10000):
        masslog = log(mass * density)
        total += masslog/(i+1)
    return total

我很惊讶map事情的速度如此之快,所以...

def groffle_faster(mass, density):
    total = 0.0
    masslog = log(mass * density)
    return map(sum, (masslog/(i+1) for i in range(10000)))

查看执行时间的差异,没有可比性。groffle_faster() 更快,但它返回一个地图对象。地图对象应包含作为浮点数的总和。

无论如何,我可以将浮动从地图对象中取出?

谢谢!

4

1 回答 1

3

它更快,因为它什么都不做;如果是这样,它就行不通了。

>>> mass = 1.2
>>> density = 2.3
>>> masslog = math.log(mass * density)
>>> map(sum, (masslog/(i+1) for i in range(10000)))
<map object at 0x7feccaf1fc18>

这个map对象是一个惰性对象,它将产生sum应用于迭代的每个元素的函数的结果,在这种情况下是生成器表达式(masslog/(i+1) for i in range(10000))。没有进行任何计算。

但这无论如何在这里没有多大意义,因为您正在尝试将函数单独sum应用于每个元素,因此:

>>> list(map(sum, (masslog/(i+1) for i in range(10000))))
Traceback (most recent call last):
  File "<ipython-input-13-c0f9c805843a>", line 1, in <module>
    list(map(sum, (masslog/(i+1) for i in range(10000))))
TypeError: 'float' object is not iterable

你真正想要的只是

>>> sum(masslog/(i+1) for i in range(10000))
9.936677928893602

这会给

>>> %timeit groffle_slow(1.5, 2.5)
100 loops, best of 3: 5.08 ms per loop
>>> %timeit groffle_fast(1.5, 2.5)
100 loops, best of 3: 3.02 ms per loop

但既然sum(1/(i+1) for i in range(10000))是一个固定的数字,我不知道你为什么不简单地使用类似的东西

>>> def groffle_O1(mass, density):
...     MSUM = 9.787606036044345
...     return log(mass*density) * MSUM
... 
>>> %timeit groffle_O1(1.5, 2.5)
1000000 loops, best of 3: 424 ns per loop

但是您还没有指定您真正在哪些约束条件下运行,因此很难知道您的真正问题是什么。

于 2015-07-22T00:19:54.537 回答