Numpy Get Shape Of Array In Python Code