4
$\begingroup$

I know that there has been an answer on meshgrid function of MATLAB. For functions involving more than 3 variables, this function does not work. In MATLAB, the function ndgrid works for 1 to n-dimensions and is similar to meshgrid. I just wanted to implement ndgrid in Mathematica so that I don't have to loop over other variables during evaluation of certain codes.

$\endgroup$
  • $\begingroup$ Yes, that's a good idea. By all means go ahead, and report back when you have made some progress :-) $\endgroup$ – MarcoB Feb 16 at 23:32
8
$\begingroup$

This might do it, if I understood the MATLAB docs:

ndgrid[args___] /; AllTrue[{args}, VectorQ] :=
  Table[
   With[{aa = {args}[[i]]},
    ArrayPad[#, ReplacePart[{0, Length[#] - 1} & /@ {args}, {i, 2} -> 0], #] &@
     ArrayReshape[aa, ReplacePart[ConstantArray[1, Length@{args}], i -> Length@aa]]
    ],
   {i, Length@{args}}];

xa = {1, 3, 5, 7, 9, 11, 13, 15, 17, 19};
ya = {2, 4, 6, 8, 10, 12};

MatrixForm /@ ndgrid[xa, ya]

Mathematica graphics

MatrixForm /@ ({xx, yy, zz} = ndgrid[{1, 2, 3}, {4, 5}, {6, 7, 8, 9}])

Mathematica graphics

Check with MATLink:

Needs["MATLink`"]
OpenMATLAB[]

MEvaluate["[X,Y,Z]=ndgrid([1 2 3],[4 5],[6 7 8 9])"];
{MGet["X"], MGet["Y"], MGet["Z"]} == {xx, yy, zz}
(*  True  *)
$\endgroup$
6
$\begingroup$

Here's my implementation:

Clear[ndgridx];
ndgridx[a__?VectorQ] := Transpose[Outer[List, a, 1], Range[1 + Length@{a}]~RotateLeft~1]

If you also need 2nd syntax i.e.

[X1,X2,...,Xn] = ndgrid(xg)

then

Clear[ndgridx];
ndgridx /: (lhs_List = ndgridx[a_]) := lhs = ndgridx @@ ConstantArray[a, Length@lhs]
ndgridx /: (lhs_List = ndgridx[a__?VectorQ]) := 
 lhs = Transpose[Outer[List, a, 1], Range[1 + Length@{a}]~RotateLeft~1]

Notice in this case single ndgridx[……] no longer works, it always needs a left hand side that is a List. Test:

{testx, testy} = ndgridx[{1, 2, 4}];

MatrixForm@testx

Mathematica graphics

MatrixForm@testy

Mathematica graphics

range = {min, max} = {-1, 1};

{xm, ym, zm} = ndgridx@Range[min, max, 0.01];

ListDensityPlot3D[Exp[-(xm^2 + zm^2 + ym^2)], DataRange -> {range, range, range}]

Mathematica graphics

$\endgroup$

Your Answer

By clicking “Post Your Answer”, you agree to our terms of service, privacy policy and cookie policy

Not the answer you're looking for? Browse other questions tagged or ask your own question.