python随机以概率从一个集合中选择元素 随机选取元素

Python随机概率函数用法

numpy.random.choice(a, size=None, replace=True, p=None)
第一个参数是值或数组 第二个是返回个数 第三个是值选取概率
参数的含义分别是，从a中以概率随机选择size数量的元素。a可以是整数或数组等
replace代表选择的值是否放回去， False的话，那么出来的数都不一样，True是放回去，出现的值有可能是重复的

下方是从-1 和 1 中随机选择一个，1出现的概率是0.6 -1出现的概率是0.4

import numpy as np
print np.random.choice([1,-1], 1,replace=True, p=[0.6, 0.4]) 

选取3个随机整数

np.random.choice(5, 3) 
array([0, 3, 4])

按概率选择3个整数

np.random.choice(5, 3, p=[0.1, 0, 0.3, 0.6, 0]) 
array([3, 3, 0])

更多详情参见：https://docs.scipy.org/doc/numpy/reference/generated/numpy.random.choice.html