Return Dimensions Of Array Python