how to make 3d shapes with python matplotlib