Python随机选择一部分训练样本作为测试样本

版权声明：本文为博主原创文章，转载请附上博文链接！ https://blog.csdn.net/weixin_38314865/article/details/84198005

假设训练样本有30个，从训练样本中随机获得10个作为测试样本，剩下20个继续作为训练样本

import numpy as np

trainingSet = list(range(30)) # 训练样本下标
testSet = []           
for i in range(10):
    randIndex = int(np.random.uniform(0, len(trainingSet))) # 获得0~len(trainingSet)的一个随机数
    testSet.append(trainingSet[randIndex])
    del(trainingSet[randIndex])

print(trainingSet) #[0, 2, 3, 4, 5, 6, 7, 9, 10, 11, 12, 13, 15, 16, 17, 19, 23, 24, 25, 27]
print(testSet) #[22, 14, 21, 29, 8, 20, 26, 28, 1, 18]