相关参数
- size:从元组,列表,数组中取多少个数据,不写,默认为1个
- replace:是否可以取相同元素:True:可以,False:不可以,默认是True
- p:实际是个数组,大小与数组相同,用来规定选取数组中每个元素的概率,默认为概率相同。注意:概率相加等于1
A = np.array([[0, 1, 2],
[3, 4, 5],
[6, 7, 8],
[9, 7, 6],
[3, 2, 2],
[0, 1, 0],
[1, 3, 1],
[0, 4, 1],
[2, 4, 2],
[3, 6, 1]])
B = A[np.random.choice(A.shape[0], size=4, replace=False), :]
print(B)
输出
[[0 1 0]
[3 6 1]
[6 7 8]
[0 1 2]]