以一定的概率生成数据集

Question

假设我有 60% 的红球和 40% 的蓝球。如何生成 1000 个数据集，每个数据集包含 10 个球？ (Python)

Answer 1

您可以使用 random.choices 根据提供的重量从 collection 中随机挑选物品。

dataset = random.choices(
    population=['blue', 'red'],
    weights=[0.4, 0.6],
    k=10
)

示例输出： ['red', 'blue', 'blue', 'blue', 'blue', 'blue', 'red', 'red', 'blue', 'red']

Generating datasets with a certain probability