numpy随机选择函数

numpy.random.choise(a, size=None, replace=True, p=None)

a : 一维列表或者整数， 表示抽取的列表。

size : 整数或者元组，抽取的矩阵形状

repalce : 是否放回， TRUE表示放回。FALSE表示不返回。

p : 随机选择的概率分布

import numpy as npnp.random.choice(3, size=(2, 3), p=(0.2, 0.2, 0.6))'''
array([[2, 0, 2],[2, 2, 2]])
'''np.random.choice([2, 3, 4], size=(2, 3), p=(0.2, 0.2, 0.6))'''
[[2 4 3][4 4 4]]
'''