python numpy-shape函数

以元组形式返回数组各个维度的元素个数

一维数组

group = array([1, 1, 0, 0])
print(group)
print(group.shape)

输出

[1 1 0 0]
(4,)

二维数组

group = array([[1.0, 1.1], [1.0, 1.0], [0, 0], [0, 0.1]])
print(group)
print(group.shape)

输出

[[ 1.0, 1.1]
 [ 1.0, 1.0]
 [ 0, 0]
 [ 0, 0.1]]
(4, 2)

通过返回的元组，可看出这个二维数组中，一维是4个元素，二维是2个元素。

三维数组

c = array([[[0, 1, 2, 0, 1, 2]],
           [[0, 1, 2, 0, 1, 2]]])
print(c)
print(c.shape)

输出

[[[0, 1, 2, 0, 1, 2]],
 [[0, 1, 2, 0, 1, 2]]]
(2, 1, 6)

通过返回的元组，可知三位数组第一维有2个元素，第二维有1个元素，第三维有6个元素

很多时候还会直接取shape得到的元组的元素，如

gourp.shape[0]