以随机顺序创建具有相等数量 0 和 1 的数组？

Question

在我解释之前，这是我的代码供参考：

import numpy as np 

arrayteam = [[3,3,3,3,3,3],[2,2,2,2,2,2],[1,1,1,1,1,1]]
#nteams = 3
#nsubteam = 2

#newnums = np.zeros((len(arrayteam),len(arrayteam[0])))

subteam = [] 
for i in range(len(arrayteam)):
    subteam.append(np.random.choice([0,1],size=len(arrayteam[i]),p=[0.5,0.5]))

print (subteam)

这是输出：

[array([0, 1, 0, 1, 1, 1]), array([1, 1, 0, 1, 0, 0]), array([1, 0, 1, 1, 1, 1])]

如您所见，它随机选择我想要的 0 和 1，但是每个数组中 0 和 1 的数量不相等，显然是因为我将它设置为 p=0.5，所以有 50% 的机会将选择 0 或 1。我想拥有它，以便每个数组中有 3 个零和 3 个 1，但它们以随机顺序出现。我该怎么做？

另外，我怎样才能最终改变我希望零和一出现的位置？例如，如果我希望数组中的前 3 个数字为 0，而后 3 个数字为 1，该怎么办？或者如果我想让他们交替使用呢？

Answer 1

使用random.shuffle。请注意，它会就地洗牌，并返回None：

import random
length = 3
subteam = [0] * length + [1] * length
random.shuffle(subteam)
print(subteam)