您的位置:首页 > 编程语言 > Python开发

numpy函数向量化,np.vectorize

2021-03-12 17:29 106 查看

 

import numpy as np
import time

def myfunc(a, b):
if a>b:
return a-b
else:
return a+b

vfunc = np.vectorize(myfunc)
t0=time.time()
vfunc(np.random.randint(5,size=10000000), 2)
t1=time.time()
print('speed_times:'+str(t1-t0))
#speed_times:3.2735631465911865
t0=time.time()
[myfunc(data, 2) for data in np.random.randint(5,size=10000000)]
t1=time.time()
print('speed_times:'+str(t1-t0))

 

结果:向量化后的并行运算速度要比普通的执行速度快很多

speed_times:2.7337491512298584
speed_times:9.00490427017212

 

内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: