在编写脚本时,我发现了numpy.random.choice函数。我实现它是因为它比等价的if语句干净得多。但是,在运行脚本之后,我意识到它比if语句慢得多。
以下是MWE。第一种方法需要0.0秒,而第二种方法需要7.2秒。如果放大i循环,您将看到random.choice减慢的速度有多快。
有人能解释为什么随机选择要慢得多吗?import numpy as np
import numpy.random as rand
import time as tm
#-------------------------------------------------------------------------------
tStart = tm.time()
for i in xrange(100):
for j in xrange(1000):
tmp = rand.rand()
if tmp < 0.25:
var = 1
elif tmp < 0.5:
var = -1
print('Time: %.1f s' %(tm.time() - tStart))
#-------------------------------------------------------------------------------
tStart = tm.time()
for i in xrange(100):
for j in xrange(1000):
var = rand.choice([-1, 0, 1], p = [0.25, 0.5, 0.25])
print('Time: %.1f s' %(tm.time() - tStart))