diff options
-rw-r--r-- | utils/sddm/renormalize.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/utils/sddm/renormalize.py b/utils/sddm/renormalize.py index f39ee91..5a43349 100644 --- a/utils/sddm/renormalize.py +++ b/utils/sddm/renormalize.py @@ -40,17 +40,17 @@ def read_mcpl(filename): data = np.genfromtxt(filename,skip_header=1) rv = {} n = 1 - i = 1 + i = 0 with gzip.open(filename,"r") as f: lines = f.readlines() nevents = int(lines[0].split()[0]) while True: - nparticles = int(data[i-1,0]) - rv[n] = np.genfromtxt(lines[i:i+nparticles]) + nparticles = int(data[i,0]) + rv[n] = data[i:i+nparticles] n += 1 i += nparticles - if i - 1 >= data.shape[0]: + if i >= data.shape[0]: break assert nevents == len(rv) |