selection = sel) dt_corsika = root2array(filenames = f_corsika, branches=variables + weightvars, treename = "tree", selection = sel+"&&primPDG >= 1000260530") #-----------------------------------------# # Calculate the weights #-----------------------------------------# wt = WeightTool() w_nugen = np.zeros(len(dt_nugen),dtype=float) for i in range(len(dt_nugen)): w_nugen[i] = wt.getE2(dt_nugen[i]) w_corsika = np.zeros(len(dt_corsika),dtype=float) for i in range(len(dt_corsika)): w_corsika[i] = wt.getCorsika(dt_corsika[i]) #w_nugen = w_nugen.reshape((len(w_nugen),1)) #w_nugen = pd.DataFrame(w_nugen,columns=['w']) #-----------------------------------------# # Merge data #-----------------------------------------# dt_nugen = append_fields(base = dt_nugen, names = 'we2', data = w_nugen,