Numpy meshgrid in 3D

Question

Numpy\'s meshgrid is very useful for converting two vectors to a coordinate grid. What is the easiest way to extend this to three dimensions? So given three vectors x, y, an

Answer 1

Here is a multidimensional version of meshgrid that I wrote:

def ndmesh(*args):
   args = map(np.asarray,args)
   return np.broadcast_arrays(*[x[(slice(None),)+(None,)*i] for i, x in enumerate(args)])

Note that the returned arrays are views of the original array data, so changing the original arrays will affect the coordinate arrays.