Making 3d Shapes In Numpy