numpy 中的 shape 方法返回的是一个数组的大小。
我们定义了一个 numpy 数组:
from numpy import *
group = array([[1.0, 1.1], [1.0, 1.0], [0, 0], [0, 0.1]])
使用 shape 方法查看该数组的大小:
group.shape
得到结果 (4, 2)
shape 方法还可以按维度进行查看,我们来查看该数组的第一维大小:
group.shape[0]
得到结果 4
查看数组第二维大小:
group.shape[1]
得到结果 2