python - 将集合划分为元素数量相等的子集

Question

为了进行心理实验，我必须将由 4 个特征(实数)描述的一组图片 (240) 分成 3 个子集，每个子​​集 (240/3 = 80) 中的元素数量相等一种所有子集在这些特征方面近似平衡的方式(在均值和标准差方面)。

有人可以推荐一种算法来实现自动化吗？我可以使用 Python 或 R 中的任何包/模块来做到这一点吗？我应该从哪里开始？

Answer 1

如果我理解正确你的问题，你可以在 python 中使用 random.sample():

import random

pool = set(["foo", "bar", "baz", "123", "456", "789"]) # your 240 elements here
slen = len(pool) / 3 # we need 3 subsets
set1 = set(random.sample(pool, slen)) # 1st random subset
pool -= set1
set2 = set(random.sample(pool, slen)) # 2nd random subset
pool -= set2
set3 = pool # 3rd random subset