python numpy 长度
时间: 2023-10-05 13:04:31 浏览: 113
Python numpy
在使用NumPy库时,可以使用`len()`函数获取NumPy数组的长度。然而,对于多维数组,`len()`函数返回的是数组的第一个维度的长度,而不是整个数组的长度。
如果要获取整个NumPy数组的长度(即元素的总数),可以使用`numpy.size()`函数或者通过查看数组的`shape`属性来获取。
以下是示例代码:
```python
import numpy as np
# 创建一个一维数组
arr1 = np.array([1, 2, 3, 4, 5])
print("arr1长度:", len(arr1))
print("arr1大小:", np.size(arr1))
# 创建一个二维数组
arr2 = np.array([[1, 2, 3], [4, 5, 6]])
print("arr2长度:", len(arr2))
print("arr2大小:", np.size(arr2))
print("arr2形状:", arr2.shape)
```
输出示例:
```
arr1长度: 5
arr1大小: 5
arr2长度: 2
arr2大小: 6
arr2形状: (2, 3)
```
注意,对于多维数组,`shape`属性返回一个元组,其中的每个元素表示对应维度的长度。在上面的例子中,`arr2.shape`返回的是`(2, 3)`,表示该二维数组有2行和3列。
阅读全文