物理のベンチ by mitta

学んだことを発信します。備忘録も書きます。間違いがあればコメントください。

NumPyを使ってCERN ROOTの解析用ダミーデータを生成する1-ランダムデータ編

やること

ダミーのテキストデータをNumPyで作る。
色々な形状な分布を生成して後々ROOTで解析する。

コード

import numpy as np

kNumEvent = 1000
fout = open("output.dat","w")
for i in range(kNumEvent):
    rand1 = round(np.random.normal(0, 1),5) # 正規分布(平均, 標準偏差) 
    rand2= round(np.random.rand(),5) # 一様分布()
    rand3 = round(np.random.binomial(100,0.3),5) # 二項分布(試行回数, 成功確率)
    rand4 = round(np.random.beta(2,5),5) # β分布(α値, β値)
    rand5 = round(np.random.gamma(5,1),5) #γ分布(形状母数(k), 尺度母数(Θ))
    rand6 = round(np.random.chisquare(3),5) # χ二乗分布(自由度(k))   
    fout.write(str(rand1)+" "+str(rand2)+" "+str(rand3)+" "+str(rand4)+" "+str(rand5)+" "+str(rand6)+"\n")
fout.close()

print("F!!!!!")