The interface between PYTHIA and NumPy
Project description
numpythia provides an interface between PYTHIA and NumPy allowing you to generate events as NumPy arrays of particle four-momenta. By default numpythia only depends on NumPy and builds internal copies of the PYTHIA and HepMC source code.
from numpythia import generate, STATUS, HAS_END_VERTEX, ABS_PDG_ID
from numpythia.testcmnd import get_cmnd
for event in generate(
get_cmnd('w'), events=1,
find=(STATUS == 1) & ~HAS_END_VERTEX &
(ABS_PDG_ID != 12) & (ABS_PDG_ID != 14) & (ABS_PDG_ID != 16),
random_state=1):
print(event)
The dtype of each generated event is:
np.dtype([('E', 'f8'), ('px', 'f8'), ('py', 'f8'), ('pz', 'f8'), ('mass', 'f8'),
('prodx', 'f8'), ('prody', 'f8'), ('prodz', 'f8'), ('prodt', 'f8'),
('pdgid', 'i4'), ('status', 'i4')])
Also see pyjet for jet clustering.
Standalone Installation
To simply use the built-in PYTHIA and HepMC:
pip install --user -v numpythia
And you’re good to go!
Support for building against an external PYTHIA, Delphes, and much more will come soon.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
numpythia-0.1.0.tar.gz
(6.5 MB
view hashes)