Numpy Get Shape Of Array Python Question