From 00a64e94c87fd9ccd1a0c3b01f49832990139dc1 Mon Sep 17 00:00:00 2001 From: tlatorre Date: Sun, 3 Jan 2021 11:41:43 -0600 Subject: speed up read_mcpl() --- utils/sddm/renormalize.py | 8 ++++---- 1 file 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) -- cgit