建立在numpy函數中以隨機獲得索引

Question

是否存在numpy功能以返回索引idx來自一維數組r，概率爲r[i]。所有r[i]非負數和總和爲一。

Answer 1

幾乎建於：

# setup (do this once) 
cs = r.cumsum() 

# get a random value with probability r[i] 
rn = np.random.uniform() 
idx = cs.searchsorted(rn) 

Answer 2

scipy.stats可以幫助，即使它不是在與NumPy的要求：

import scipy.stats 
r = [.1,.8,.1] 
x = range(len(r)) 
mydist = scipy.stats.rv_discrete(values=(x,r), name='mydist') 
print mydist.rvs(size=20) 
# array([1, 1, 0, 0, 1, 1, 2, 2, 1, 1, 2, 1, 2, 1, 1, 1, 1, 1, 1, 1])