Commit 86eb2f82 authored by Dennis Noll's avatar Dennis Noll
Browse files

[utils] coffea: naming of vars in tree

parent 48b68b8d
......@@ -695,13 +695,13 @@ class TreeExporter(ArrayExporter):
parts = [-1] if len(var_array.shape) == 2 else range(var_array.shape[1])
for i in parts:
if i == -1:
name = var
name = ""
vals = var_array
else:
name = f"{var}{i}"
name = f"{var}{i}_"
vals = var_array[:, i]
for j in range(vals.shape[-1]):
var_name = f"{name}_{vars[var][2][j]}"
var_name = f"{name}{vars[var][2][j]}"
v = vals[:, j]
tree[var_name].append(v)
for cat_flag in arrays.keys():
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment