Numpy 操作trick

最新推荐文章于 2022-03-03 22:30:47 发布

沧海二阳

最新推荐文章于 2022-03-03 22:30:47 发布

阅读量103

点赞数

分类专栏：数据处理

本文链接：https://blog.csdn.net/weixin_39451323/article/details/103747485

版权

数据处理专栏收录该内容

11 篇文章 0 订阅

订阅专栏

Numpy Trick

- 1.扩张数组维度

1.扩张数组维度

s = 'yyyyy 112 113 114 115'
anno = s.strip().split(" ")
bbox = list(map(float,anno[1:]))
boxes1 = np.array(bbox)[None] #单纯扩张维度
boxes2 = np.array(bbox, dtype=np.float32).reshape(-1,4) #4列，行自动生成
boxes3 = np.array(bbox, dtype=np.float32).reshape(1,-1) #1行， 列自动生成
print("bbox:{} type：{}\nboxes1:{} shape:{}\nboxes2:{} shape:{}\nboxes3:{} shape:{}".format(bbox,type(bbox), boxes1,boxes1.shape,boxes2,boxes2.shape,boxes3,boxes3.shape))boxes1,boxes1.shape,boxes2,boxes2.shape,boxes3,boxes3.shape))

结果：

bbox:[112.0, 113.0, 114.0, 115.0] type：<class 'list'>
boxes1:[[112. 113. 114. 115.]] shape:(1, 4)
boxes2:[[112. 113. 114. 115.]] shape:(1, 4)
boxes3:[[112. 113. 114. 115.]] shape:(1, 4)