I will explain how to use numpy mgrid, a powerful tool for creating multidimensional grids.
Numpy mgrid is a function that returns a dense multi-dimensional “meshgrid”. A meshgrid is an array that contains the coordinates of a rectangular grid.
Online How to Python stuff