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,
usemask = False,
dtypes=float)
