IAP GITLAB

Skip to content
Snippets Groups Projects
Commit ac8656d5 authored by Maximilian Reininghaus's avatar Maximilian Reininghaus :vulcan:
Browse files

even better flow bin handling

parent ccd7e6e8
No related branches found
No related tags found
No related merge requests found
import numpy as np
import matplotlib.pyplot as plt
import boost_histogram as bh
import operator
import functools
def read_hist(filename):
"""
read numpy file produced with CORSIKA 8's save_hist() function into
boost-histogram object.
"""
d = np.load(filename)
axistypes = d['axistypes'].view('c')
overflow = d['overflow']
underflow = d['underflow']
axes = []
for i, (at, has_overflow, has_underflow) in enumerate(zip(axistypes, overflow, underflow)):
if at == b'c':
axes.append(bh.axis.Variable(d[f'binedges_{i}'], overflow=has_overflow, underflow=has_underflow))
elif at == b'd':
axes.append(bh.axis.IntCategory(d[f'bins_{i}'], growth=(not has_overflow)))
h = bh.Histogram(*axes)
h.view(flow=True)[:] = d['data']
return h
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment