Get Shape Of Array Python Size Function