Numpy：如何将一个矩阵随机拆分/选择成n个不同的矩阵

Question

• 我有一个核矩阵的形状（460158）。
• 我想按60%，20%，20%的行数随机拆分矩阵
• 这是我需要的机器学习任务
• 是否有随机选择行的numpy函数？

Answer 1

如果您想用相同的第一维连续地洗牌多个数组x，y，z，HYRY的答案的一个补充：x.shape[0] == y.shape[0] == z.shape[0] == n_samples。

你可以：

rng = np.random.RandomState(42)  # reproducible results with a fixed seed
indices = np.arange(n_samples)
rng.shuffle(indices)
x_shuffled = x[indices]
y_shuffled = y[indices]
z_shuffled = z[indices]

然后像海瑞的回答那样，对每个洗牌数组进行分割。