如何知道numpy数组的维度和大小
时间: 2024-05-06 10:19:34 浏览: 67
Numpy数组操作
可以使用numpy数组的属性来获取数组的维度和大小。
1. 维度:使用数组的ndim属性可以获取数组的维度数。例如:
```python
import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6]])
print(arr.ndim) # 输出 2
```
2. 大小:使用数组的shape属性可以获取数组的大小,即每个维度上的元素个数。例如:
```python
import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6]])
print(arr.shape) # 输出 (2, 3)
```
上述代码中,数组arr有2个维度,每个维度上的元素个数分别是2和3,因此数组大小为(2, 3)。
阅读全文