generalize function.asarray for varying ndim args

evalf · Feb 27, 2015 · ee5ef7d · ee5ef7d
1 parent acd890f
commit ee5ef7d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/nutils/function.py b/nutils/function.py
@@ -2347,6 +2347,8 @@ def asarray( arg ):
     return array
 
   args = [ asarray(a) for a in arg ]
+  ndim = _max( arg.ndim for arg in args )
+  args = [ arg[(_,)*(ndim-arg.ndim)] for arg in args ]
 
   if all( isinstance( arg, numpy.ndarray ) for arg in args ):
     array = numpy.array( args )