|
12 | 12 | # PLOTTING FUNCTION # |
13 | 13 | ##################### |
14 | 14 |
|
15 | | -def scatter3d(x, y, z, values, jets=None, eventVector=None, size=25, colorsMap='jet', saveName='Plots/Test.pdf'): |
| 15 | +def scatter3d(x, y, z, values, jets=None, eventVector=None, size=25, colorsMap='jet', saveName='Plots/Test.eps'): |
16 | 16 |
|
17 | 17 | cm = plt.get_cmap(colorsMap) # set up color map |
18 | 18 | cNorm = matplotlib.colors.Normalize(vmin=min(values), vmax=max(values)) # normalize to min and max of data |
@@ -89,20 +89,20 @@ def scatter3d(x, y, z, values, jets=None, eventVector=None, size=25, colorsMap=' |
89 | 89 | jet2 = data['N_Subjettiness/bestJets2'] |
90 | 90 | x, y, z = np.nonzero(ECAL[eventN]) # first event |
91 | 91 | E = ECAL[eventN][np.nonzero(ECAL[eventN])] |
92 | | - if len(E)>0: scatter3d(x, y, z, E, (jet1[eventN], jet2[eventN]), eventVector, size=25, saveName=saveDirectory+"ECAL_event"+str(eventN)+".pdf") |
| 92 | + if len(E)>0: scatter3d(x, y, z, E, (jet1[eventN], jet2[eventN]), eventVector, size=25, saveName=saveDirectory+"ECAL_event"+str(eventN)+".eps") |
93 | 93 |
|
94 | 94 | # HCAL = data['HCAL/HCAL'] |
95 | 95 | # x, y, z = np.nonzero(HCAL[eventN]) # first event |
96 | 96 | # E = HCAL[eventN][np.nonzero(HCAL[eventN])] |
97 | | - # if len(E)>0: scatter3d(x, y , z, E, size=5, saveName=saveDirectory+"HCAL_event"+str(eventN)+".pdf") |
| 97 | + # if len(E)>0: scatter3d(x, y , z, E, size=5, saveName=saveDirectory+"HCAL_event"+str(eventN)+".eps") |
98 | 98 |
|
99 | 99 | # # plot averages |
100 | 100 | # ECAL_average = np.average(data['ECAL/ECAL'], axis=0) |
101 | 101 | # x, y, z = np.nonzero(ECAL_average) # first event |
102 | 102 | # E = ECAL_average[np.nonzero(ECAL_average)] |
103 | | -# scatter3d(x, y, z, E, size=25, saveName=saveDirectory+"ECAL_average.pdf") |
| 103 | +# scatter3d(x, y, z, E, size=25, saveName=saveDirectory+"ECAL_average.eps") |
104 | 104 |
|
105 | 105 | # HCAL_average = np.average(data['HCAL/HCAL'], axis=0) |
106 | 106 | # x, y, z = np.nonzero(HCAL_average) # first event |
107 | 107 | # E = HCAL_average[np.nonzero(HCAL_average)] |
108 | | -# scatter3d(x, y , z, E, size=5, saveName=saveDirectory+"HCAL_average.pdf") |
| 108 | +# scatter3d(x, y , z, E, size=5, saveName=saveDirectory+"HCAL_average.eps") |
0 commit comments