diff --git a/scripts/checkPulseHam_diff.py b/scripts/checkPulseHam_diff.py index d639f11..e40f404 100644 --- a/scripts/checkPulseHam_diff.py +++ b/scripts/checkPulseHam_diff.py @@ -3,7 +3,7 @@ from ROOT import TFile, TLorentzVector, TVector3, TRotation, TLorentzRotation, TMath, TH1D, TCanvas, TH2D, TObject, TF1 import matplotlib.pyplot as plt import pandas as pd - +import pickle macros = '/home/hep/matzeni/CCE/macros/CCEScan/' filelumi = TFile('../data/db/lumi.root') @@ -48,6 +48,8 @@ d = {'Fill' : pd.Series(vd_fill), 'Time' : pd.Series(vd_time)} df =pd.DataFrame(d) +ACHT =[] + for isect in sectors: print'=====================================================' @@ -104,15 +106,31 @@ for i in xrange(len(vD_time)): print(i) if (vDiff[i] >= 3*vData_err[i] ): - print 'WARNING! Check Sector: {} for Fill: {} '.format(isect,df['Fill'][df.Time==vD_time[i]]) + #print 'WARNING! Check Sector: {} for Fill: {} '.format(isect,df['Fill'][df.Time==vD_time[i]].values[0]) + ACHT = np.append([isect,int(df['Fill'][df.Time==vD_time[i]].values[0])],ACHT) +# print(ACHT) + + #plt.plot(vH_time,vH_V, 'r--',vD_time,vH_inter, 'bs',vD_time,vData, 'gs') + #plt.show() + #raw_input('>') + +ACHT = np.reshape(ACHT,(len(ACHT)/2,2)) +print 'ACHT',ACHT + +data_frame = pickle.load(open('monotonics.pkl', 'rb')) +#ACHT_ele = data_frame[['Sector','Fill']][data_frame['V bias'] < 125].values +ACHT_ele = data_frame[['Sector','Fill']].values - plt.plot(vH_time,vH_V, 'r--',vD_time,vH_inter, 'bs',vD_time,vData, 'gs') - plt.show() - raw_input('>') +print 'ACHT_ele', ACHT_ele - ''' +aset = set([tuple(x) for x in ACHT]) +bset = set([tuple(x) for x in ACHT_ele]) +inter = np.array([x for x in aset & bset]) +print 'Intersection', inter + +''' a = [ifill, vData_t[0]] @@ -122,7 +140,7 @@ print 'vH_V: ',vH_V print 'vd_fill: ',vd_fill - + print 'vd_time: ',vd_time #Time used to do the Hamburg simulation