将列表随机划分为两个互补的子列表

Question

我有一个列表，我想将它随机分成两个已知大小的子列表，它们是相互补充的。例如，我有[1,5,6,8,9]，我想将其划分为[1,5,9] 和[6,8]。我不太关心效率，只希望它工作。顺序无关紧要。

我开始：

pop = [...] #some input
samp1 = random.sample(pop, samp1len)
samp2 = [x for x in pop if x not in samp1]

但是，此解决方案因重复项目而失败。如果pop = [0,0,0,3,5]，并且第一个选择的长度为3 是[0,3,5]，我仍然希望samp2 是[0,0]，我的代码目前无法提供。

是否有一些我错过的随机内置选项？谁能提供一个简单的解决方案？

Answer 1

这样的事情怎么样？

生成索引列表并打乱它们：

>>> indices = range(len(pop))
>>> random.shuffle(indices)

然后对索引列表进行切片并使用operator.itemegetter 获取项目：

>>> from operator import itemgetter
>>> itemgetter(*indices[:3])(pop)
(0, 0, 3)
>>> itemgetter(*indices[3:])(pop)
(5, 0)