python - 如何创建特定大小的二项式数组

Question

我正在尝试生成一个长度为 100 的 numpy 数组，随机填充 5 个 1 和 0 的集合，如下所示:

    [ [1,1,1,1,1] , [0,0,0,0,0] , [0,0,0,0,0] ... [1,1,1,1,1], [0,0,0,0,0] ]

本质上，每个位置应该有 50% 的机会出现 5 个 1，并且有 50% 的机会出现 5 个 0

目前，我一直在搞乱 numpy.random.binomial()，并尝试运行:

    numpy.random.binomial(1, .5 , (100,5))

但这会创建一个这样的数组:

    [ [0,1,0,0,1] , [0,1,1,1,0] , [1,1,0,0,1] ... ]

我需要每组元素保持一致而不是随机的。我怎样才能做到这一点？

Answer 1

使用numpy.random.randint生成 100 个 1 和 0 的随机列，然后使用 tile重复该列 5 次:

>>> numpy.tile(numpy.random.randint(0, 2, size=(100, 1)), 5)
array([[0, 0, 0, 0, 0],
       [0, 0, 0, 0, 0],
       [1, 1, 1, 1, 1],
       [0, 0, 0, 0, 0],
       [1, 1, 1, 1, 1],
       [1, 1, 1, 1, 1],
       [0, 0, 0, 0, 0],
...