Python中的 numpy.alen()

简介

numpy.alen()函数返回给定数组的长度或形状的第一个元素的长度。与len()函数不同，它可以处理多维数组。

语法

numpy.alen(arr)

参数

• arr: 输入数组

返回值

返回数组的长度或形状的第一个元素的长度。

示例

import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6]])
print(np.alen(arr)) # 2
print(np.alen(arr[0])) # 3
print(np.alen(arr[:,0])) # 2
print(np.alen(arr[0,:2])) # 2

以上代码将输出：

2
3
2
2

说明

对于一维数组，alen()函数返回其长度。

对于多维数组，alen()函数返回第一维的长度。

对于形状为(a,b,...)的多维数组，alen()函数返回a。如果数组的形状为()，alen()函数返回0。