How to generate meshgrid in 3D with Numpy?

Let’s learn a very interesting trick: how to generate meshgrid in 3D with Numpy.

Generating three arrays

First I am generating 3 arrays:

import numpy as np

xs = np.linspace(0., 1., 2)
ys = np.linspace(1., 2., 2)
zs = np.linspace(3., 4., 2)

Obviously, they look like this:

import numpy as np

xs = np.linspace(0., 1., 2)
ys = np.linspace(1., 2., 2)
zs = np.linspace(3., 4., 2)

print(f"X values: \n {xs}")
print(f"Y values: \n {ys}")
print(f"Z values: \n {zs}")

Generating 3D meshgrid arrays

To generate a 3D meshgrid array, I used the meshgrid function and provided the created arrays as parameters.

import numpy as np

xs = np.linspace(0., 1., 2)
ys = np.linspace(1., 2., 2)
zs = np.linspace(3., 4., 2)

meshgrid_array = np.meshgrid(xs, ys, zs)
print(f"My 3d meshgrid array: \n {meshgrid_array}")

This is what a 3D meshgrid array looks like.