numpy.shape()函数

最新推荐文章于 2024-08-13 10:58:13 发布

宁静致远*

最新推荐文章于 2024-08-13 10:58:13 发布

阅读量4.9k

点赞数

分类专栏：修仙之路：numpy篇

本文链接：https://blog.csdn.net/weixin_40522801/article/details/106457812

版权

修仙之路：numpy篇专栏收录该内容

18 篇文章 4 订阅

订阅专栏

shape描述的是矩阵的形状。

1 一般用法

import numpy as np

a=np.array([1,2,3])
print(a.shape）

# result
3

b=np.array([[1,2,3],[4,5,6],[7,8,9]])
print(b.shape)

# result
(3，3)

c=np.array([[[1,2,3],[4,5,6],[7,8,9]]])
print(c.shape)

# result
(1，3，3）  #(H，W，C)

2 特殊用法

import cv2

img = cv2.imread('../DAVIS-2016/JPEGImages/1080p/swing/00005.jpg')
print(img.shape)

print(img.shape[:1])  # 输出从第一项到第一项
print(img.shape[:2])  # 输出从第一项到第二项
print(img.shape[:3])  # 输出从第一项到第三项

print(img.shape[2:])  # 输出从第2+1项到最后一项
print(img.shape[1:])  # 输出从第2项到最后一项
print(img.shape[0:])  # 输出第一项到最后一项


# result

(1080, 1920, 3)

(1080,)
(1080, 1920)
(1080, 1920, 3)

(3,)
(1920, 3)
(1080, 1920, 3)