生成n倍的拆分索引

Question

我需要生成一个用于交叉验证的拆分，比如s是记录的索引

s = [1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20]

现在我想随机洗牌并将数据拆分为5倍，通常我想输出类似的东西

s = [[1 5 4 6], [2,3, 19,20], [... ], [... ], [.. ]] 

注意：在每个数组中，数字都应该是唯一的，不应该重复

我知道我可以使用chunk（），但在chunk中，你只能按顺序执行，比如1-4、5-8、，。。。。

有人能帮我吗？

Answer 1

使用random.shuffle打乱数组并将其拆分为5个部分：

对于Python2，使用

import random
s = range(1, 21)
random.shuffle(s)
s = [s[i::5] for i in range(5)]

或对于Python3:

import random
s = list(range(1, 21))
random.shuffle(s)
s = [s[i::5] for i in range(5)]

Answer 2

import random
s = [1 ,2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20]
print [random.sample(s,5) for i in xrange(len(s)/5)]