Merge 4e2e201 into d337ed0

scikit-hep · Oct 19, 2017 · 51df9c0 · 51df9c0
2 parents d337ed0 + 4e2e201
commit 51df9c0
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/root_numpy/src/tree.pyx b/root_numpy/src/tree.pyx
@@ -726,6 +726,12 @@ def array2root(arr, filename, treename='tree', mode='update'):
     # If a tree with that name exists, we want to update it
     cdef TTree* tree = <TTree*> rfile.Get(treename)
     tree = array2tree(arr, name=treename, tree=tree)
+    if "/" in treename:
+        components = treename.split("/")
+        dirs = components[:-1]
+        for d in dirs:
+            rfile.mkdir(d).cd()
+        treename = components[-1]
     tree.Write(treename, kOverwrite)
     del tree
     rfile.Close()