shape函数在numpy模块中,功能为可以读取数组或矩阵的维度大小。例如:
建立一个一维矩阵a,则shape(a)为矩阵的长度:
from numpy import *
a = array([5, 1])
print(shape(a))
>>(2,)
建立一个2*3的二维数组,b.shape[0]为矩阵的行数,b.shape[1]为矩阵的列数:
b = array([[1, 2, 3], [2, 3, 4]])
print(b.shape[0])
print(b.shape[1])
>>2
>>3
若输入为单独数值,则返回空值:
print(shape(1))
>>()