以元组形式返回数组各个维度的元素个数
一维数组
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]
网友评论