当前位置:首页 >> 脚本专栏

python怎么提高计算速度

下面是python中的一个函数计算代码:

loops=25000000

from math import*
a=range(1,loops)
def f(x):
  return 3*cos(x)+4*sin(x)**2
%timeit r=(f(x) for x in a)

效率:

1000000 loops, best of 3: 552 ns per loop

下面我们就来看一下提高计算速度的方法:

1、使用数组

import numpy as np
a = np.arange(1, loops)
%timeit r = 3 * np.cos(a) + np.sin(a)** 2

效率:

1 loop, best of 3: 3.87 s per loop

2、使用数字表达专用库numexpr

import numexpr as ne
ne.set_num_threads(1)
f = '3 * log(a) + cos(a) **2'
%timeit r = ne.evaluate(f)

效率:

1 loop, best of 3: 2.15 s per loop

3、使用多线程

ne.set_num_threads(4)
%timeit r = ne.evaluate(f)

效率:

1 loop, best of 3: 1.14 s per loop