用python概率創建二元隨機矩陣

Question

我是python的新手。我想寫一個函數產生一個隨機的nxm二進制矩陣，每個值爲0概率爲p。

我做了什麼。

def randbin(M,N,P):    # function to generate random binary matrix  
    mat = (np.random.rand(M,N)>=P).astype(int) 
    return mat 

y = randbin(5,4,0.3) 

每次打印輸出時，我都沒有按照估計的概率得到結果。我不知道我做錯了什麼。

Answer 1

我沒有看到你的方法問題...一個更好的方法來生成0和1與0的概率P的隨機矩陣是使用random.choice：

def randbin(M,N,P): 
    return np.random.choice([0, 1], size=(M,N), p=[P, 1-P]) 

爲了更好地理解，看看random.choice文檔。