Skip to content
Snippets Groups Projects

EDM4hep: mcparticles -> MCParticles

Merged Wouter Deconinck requested to merge edm4hep-mcparticles into master
3 files
+ 5
5
Compare changes
  • Side-by-side
  • Inline
Files
3
@@ -62,7 +62,7 @@ def thrown_particles_figure(rdf, save, mcbranch="MCParticles"):
get_pname = np.vectorize(lambda pid: pdgbase.GetParticle(int(pid)).GetName())
# enumerate particle names
dft.loc[:, 'pname'] = get_pname(dft['pdgID'].values)
dft.loc[:, 'pname'] = get_pname(dft['PDG'].values)
penum = {pname: i for i, pname in enumerate(dft['pname'].unique())}
dft.loc[:, 'pname_id'] = dft['pname'].map(penum)
Loading