ROOTヒストグラムにガウスシアンを当てる

Question

ROOTヒストグラムにいくつかのガウス関数を適合させるプログラムを作成しようとしていますが、残念ながら私のpyroOT経験はありません。

私はBa133の発光スペクトルのヒストグラムを持っており、検出器を較正するために、このピークのx軸値を知るためにピークにガウス分布を合わせたいと考えています。理想的には、プログラムは反復してピークを見つけることになりますが、自分で指定する必要はありません。

現在、私が持っている唯一のコードは次のとおりです。

import ROOT 

infile = ROOT.TFile("run333.root") 

Ba_spectrum = infile.Get("hQ0") 

Ba_spectrum.Draw() 

誰かが好ましくは自動、どのようにこれらのピークにガウス分布に合わせてpyrootを使用する方法を教えてくださいことができれば、それをいただければ幸いです。まともなフィット結果を得ることは、多くの場合、あなたはおそらくで始まるようにピークのおおよその位置を指定する方がいいでしょう、合理的な初期パラメータ値で始まるに依存していることを考えると

おかげ

Answer 1

。たとえば、すべての見かけのピークの高さ、平均、幅の推測を含むテキストファイルを作成することができます。

16000.0 625.0 25.0 
    500.0 750.0 50.0 
... 

次に、このようなフィットを実行します。

import ROOT 

in_file = ROOT.TFile("run333.root") 
if not in_file.IsOpen(): 
    raise SystemExit("Could not open file.") 

spectrum = in_file.Get("hQ0") 
if spectrum is None: 
    raise SystemExit("Could not find spectrum histogram.") 

N_GAUSS_PARAMS = 3 

init = [] 
with open("init.txt") as f: 
    for s in f: 
     tokens = s.split() 
     if not tokens: 
      continue 
     if len(tokens) != N_GAUSS_PARAMS: 
      raise SystemExit(
       "Bad line in initial-value file: \"{}.\"".format(s) 
      ) 

     init.append([float(t) for t in tokens]) 

n_peaks = len(init) 
n_params = N_GAUSS_PARAMS * n_peaks 

fit_function = ROOT.TF1(
    "fit_function", 
    "+".join(
     ["gaus({})".format(i) 
     for i in range(0, n_params, N_GAUSS_PARAMS)] 
    ), 0.0, 4100.0 
) 
for i in range(n_peaks): 
    for j in range(N_GAUSS_PARAMS): 
     fit_function.SetParameter(i * N_GAUSS_PARAMS + j, init[i][j]) 

spectrum.Fit(fit_function) 

for i in range(1, n_params, N_GAUSS_PARAMS): 
    print(fit_function.GetParameter(i))