np.random.choice(a, size=None, replace=True, p=None)
参数:
import numpy as np
A = np.array([1,2,3,4,5,6])
B = np.random.choice(A, 2, replace=False)
print(B)
则B的值可能为:
[1,3]
import numpy as np
A = np.array([[1,2],[3,4],[5,6],[7,2],[4,9]])
B = A[np.random.choice(A.shape[0],2),:]
则B的结果可能为:
[[4, 9],
[3, 4]]
实际上原理很简单,先生成随机索引,再把随机索引对应的二维元素取出来即可。