Python中的shape
函数是NumPy库的核心函数,用于获取数组的维度信息。以下是具体说明:
-
基本定义
shape
是NumPy数组的属性(非独立函数),用于返回数组的形状(即各维度的大小)。对于多维数组,返回值是一个元组,每个元素对应一个维度的长度。 -
返回值说明
-
一维数组 :返回一个整数,表示数组长度。
-
二维数组 :返回元组
(行数, 列数)
,例如shape
为行数,shape
为列数。 -
高维数组 :返回元组
(维度1, 维度2, ..., 维度N)
,可通过shape[-1]
获取最后一维的长度。
-
-
使用示例
import numpy as np x = np.array([[1,2,3], [4,5,6]]) print(x.shape) # 输出: (2, 3) print(x.shape) # 输出: 2(行数) print(x.shape) # 输出: 3(列数) print(x.shape[-1]) # 输出: 3(最后一维长度)
-
注意事项
-
shape
是NumPy专用函数,需先导入库(import numpy as np
)。 -
仅适用于NumPy数组,不能用于Python原生数据类型(如列表)。
-
总结 :shape
函数通过返回数组形状元组,帮助用户快速了解数据的维度结构,是进行数值计算和数据处理的基础工具。