How To Get Shape Of Array In Python