I think both of the above can be easier obtained with `np.meshgrid`, no?