NumPy更改具有給定概率的數組的元素

Question

我有一個隨機數組。我只想改變一些基於概率0.07的元素。目前我正在使用for循環遍歷每個元素。有沒有更好的方法來做到這一點？

Answer 1

如果有問題的數組被稱爲a，您可以通過

a[numpy.random.rand(*a.shape) < 0.07] 

選擇其值的0.07的平均比例，我不知道怎麼你要改變這些值。爲了將它們乘以兩，只要做

a[numpy.random.rand(*a.shape) < 0.07] *= 2.0 

Answer 2

斯文的答案是優雅的。然而，它的速度要快得多來挑選要與

n = numpy.random.binomial(len(a), 0.07) 
a[numpy.random.randint(0, len(a), size=n)] *= 2.0 

第一表達決定要多少個元素進行採樣以改變元件（n爲（a）中，但平均0.07 0和len之間的整數），第二個生成正好你想要檢索的索引數量。 （請注意，但是，你可能會得到同樣的指數幾次。）

到

的差異

a[numpy.random.rand(len(a)) < p] 

爲p接近1變小，但對於小P，它可能是10倍或更多。