如何知道numpy数组的维度和大小

可以使用numpy数组的属性来获取数组的维度和大小。

1. 维度：使用数组的ndim属性可以获取数组的维度数。例如：

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
print(arr.ndim)  # 输出 2

2. 大小：使用数组的shape属性可以获取数组的大小，即每个维度上的元素个数。例如：

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
print(arr.shape)  # 输出 (2, 3)

上述代码中，数组arr有2个维度，每个维度上的元素个数分别是2和3，因此数组大小为(2, 3)。