数据分析之np.random.choice()补充【从二维数组随机选择n行一维数组】

相关参数

• 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]]