diff options
| author | tlatorre <tlatorre@uchicago.edu> | 2021-01-03 11:41:43 -0600 |
|---|---|---|
| committer | tlatorre <tlatorre@uchicago.edu> | 2021-01-03 11:41:43 -0600 |
| commit | 00a64e94c87fd9ccd1a0c3b01f49832990139dc1 (patch) | |
| tree | fc1e48a99cb4d3163064c2d9fd1314618e0ef58e | |
| parent | c7f45ca4b215a7cf8f160cbbba7dafa96eaff228 (diff) | |
| download | sddm-00a64e94c87fd9ccd1a0c3b01f49832990139dc1.tar.gz sddm-00a64e94c87fd9ccd1a0c3b01f49832990139dc1.tar.bz2 sddm-00a64e94c87fd9ccd1a0c3b01f49832990139dc1.zip | |
speed up read_mcpl()
| -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) |