[python] -1か1を確率1/2で出力する

Question

自分が書いたコード

import random
def noise():
if random.random() >= 0.5:
return 1
else:
return -1

一様分布[0,1]からランダムに値を取り出して、0.5以上であれば1、それ以外であれば-1を返すようにしました。
何か改善点があれば教えてください。

Answer 1

基本的にはそれで良いかと。

同じロジックでも条件演算子を使うと行数が節約できます。

def noise():
    return 1 if random.random() >= 0.5 else -1

また、多数のデータを高速に生成する必要がある場合はnumpyを用いてやった方が良いでしょう。

Answer 2

参考情報

• Pythonでリストからランダムに要素を選択するchoice, sample, choices

https://note.nkmk.me/python-random-choice-sample-choices/

この記事を参考に 5 つの方法を書いてみました。

import random

random.randrange(-1, 2, 2)
random.choice([-1, 1])

int((random.randint(0, 1) - 0.5) * 2)
-1 if random.randint(0, 1) == 0 else 1
-1 if random.uniform(0, 1) < 0.5 else 1

実行例

Answer 3

random.choice([1, -1])
なんてのを見かけた。
分布に関して、未確認ｗ

Answer 4

すでに解決していますがrandom.randrange(-1,2,2)でもよさそうです。

import random
def noise():
    return random.randrange(-1,2,2) # -1以上2未満でstep=2の整数 = -1 or +1

random.seed(123)
for i in range(10):
    l = [noise() for i in range(100000)]
    print(sum(l))
"""
604
-962
518
-396
-230
-188
-530
-1488
82
-536
"""

Answer 5

x = 1 if random.random() >= 0.5 else 0