python - 创建随机 bool 数组

Question

我想在 Python 中高效地创建一个具有给定长度和给定数量的 True 值的随机 bool 数组。我找不到一个命令来执行此操作，以下是我想要的(两次)。有什么更优雅的方法吗？

import numpy as np

def randbool(length,numtrue):
    index_array=np.random.choice(length,numtrue,replace=False)
    bool_array=np.zeros(length,dtype=bool)
    bool_array[index_array]=True
    return(bool_array)

def randbool2(length,numtrue):
    bool_array=np.hstack((np.tile(True,numtrue),np.tile(False,length-numtrue)))
    np.random.shuffle(bool_array)
    return(bool_array)

print(randbool(5,2))
print(randbool2(5,2))

Answer 1

还有一个选择:

def randbool(l,n):
   return np.random.permutation(l)<n

np.random.permutation 随机排列 np.arange(l)。