3种sample整数array效率比较

method1无重复, method2/3有重复

import random
import time
import numpy as np

def method1(numsamples,batchsize,numrepeats):
    for k in range(numrepeats):
        result=random.sample(range(numsamples),batchsize)
    return True

def method2(numsamples,batchsize,numrepeats):
    for k in range(numrepeats):
        result=[random.randint(0, numsamples - 1) for _ in range(batchsize)]
    return True
def method3(numsamples,batchsize,numrepeats):
    for k in range(numrepeats):
        result=np.random.randint(0,numsamples-1,batchsize)
    return True


numsamples=1000000
batchsize=32
numrepeats=10000
starttime=time.time()
method1(numsamples,batchsize,numrepeats)
print('method1 time: {:.6f} s'.format(time.time()-starttime))
starttime=time.time()
method2(numsamples,batchsize,numrepeats)
print('method2 time: {:.6f} s'.format(time.time()-starttime))
starttime=time.time()
method3(numsamples,batchsize,numrepeats)
print('method3 time: {:.6f} s'.format(time.time()-starttime))

结果如图

method1 time: 0.343741 s
method2 time: 0.512811 s
method3 time: 0.023451 s