numpy:如何将矩阵随机拆分/选择为n个不同的矩阵 [英] Numpy: How to randomly split/select an matrix into n-different matrices

问题描述

• 我有一个形状为(4601，58)的numpy矩阵.
• 我想根据行数按60％，20％，20％的比例随机分割矩阵
• 这是我需要的机器学习任务
• 是否有一个numpy函数随机选择行?

推荐答案

您可以使用numpy.random.shuffle

you can use numpy.random.shuffle

import numpy as np

N = 4601
data = np.arange(N*58).reshape(-1, 58)
np.random.shuffle(data)

a = data[:int(N*0.6)]
b = data[int(N*0.6):int(N*0.8)]
c = data[int(N*0.8):]

这篇关于numpy:如何将矩阵随机拆分/选择为n个不同的矩阵的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！