给numpy.array增加维度的超简单方法
更新时间：2021年06月01日 14:44:42 作者：whyume
这篇文章主要介绍了给numpy.array增加维度的超简单方法，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教
输入：
import numpy as np a = np.array([1, 2, 3]) print(a)
输出结果：
array([1, 2, 3])
输入：
print(a[None])
输出结果：
array([[1, 2, 3]])
输入：
print(a[:,None])
输出结果：
array([[1],
[2],
[3]])
numpy数组的维度增减方法
使用np.expand_dims()为数组增加指定的轴,np.squeeze()将数组中的轴进行压缩减小维度。
1.增加numpy array的维度
在操作数组情况下，需要按照某个轴将不同数组的维度对齐，这时候需要为数组添加维度(特别是将二维数组变成高维张量的情况下)。
numpy提供了expand_dims()函数来为数组增加维度：
import numpy as np a = np.array([[1,2],[3,4]]) a.shape print(a) >>> """ (2L, 2L) [[1 2] [3 4]] """ # 如果需要在数组上增加维度,输入需要增添维度的轴即可，注意index从零还是 a_add_dimension = np.expand_dims(a,axis=0) a_add_dimension.shape >>> (1L, 2L, 2L) a_add_dimension2 = np.expand_dims(a,axis=-1) a_add_dimension2.shape >>> (2L, 2L, 1L) a_add_dimension3 = np.expand_dims(a,axis=1) a_add_dimension3.shape >>> (2L, 1L, 2L)
2.压缩维度移除轴
在数组中会存在很多轴只有1维的情况，可以使用squeeze函数来压缩冗余维度
b = np.array([[[[5],[6]],[[7],[8]]]])
b.shape
print(b)
>>>
"""
(1L, 2L, 2L, 1L)
array([[[[5],
[6]],
[[7],
[8]]]])
"""
b_squeeze = b.squeeze()
b_squeeze.shape
>>>(2L, 2L) #默认压缩所有为1的维度
b_squeeze0 = b.squeeze(axis=0) #调用array实例的方法
b_squeeze0.shape
>>>(2L, 2L, 1L)
b_squeeze3 = np.squeeze(b, axis=3) #调用numpy的方法
b_squeeze3.shape
>>>(1L, 2L, 2L)
以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
深入理解python中if __name__ == ‘__main__‘
很多python的文件中会有语句if __name=='__main__':,一直不太明白,最近查阅了一下资料,现在明白,本文就来深入理解一下,感兴趣的可以了解一下2023-08-08
这篇文章主要介绍了Python教程教你如何去除背景，修饰图片中的头发是设计师最烦人的任务之一,下面文章将让大家学会怎么轻松抠图去背景，需要的小伙伴可以参考一下2022-02-02
python中windows链接linux执行命令并获取执行状态的问题小结
这篇文章主要介绍了python中windows链接linux执行命令并获取执行状态,由于工具是pyqt写的所以牵扯到用python链接linux的问题，这里记录一下一些碰到的问题，需要的朋友可以参考下2022-11-11
这篇文章主要介绍了Python DES加密实现原理及实例解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-07-07
今天小编就为大家分享一篇numpy ndarray 取出满足特定条件的某些行实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-12-12
当在图像上训练深度神经网络模型时，通过对由数据增强生成的更多图像进行训练，可以使模型更好地泛化。本文将为大家介绍Python基于域相关的图像增强实现方法，需要的可以了解一下2022-01-01
python 环境搭建 及python-3.4.4的下载和安装过程
这篇文章主要介绍了python 环境搭建 python-3.4.4的下载和安装过程，文中给大家补充介绍了pycharm的基本用法，非常不错，具有一定的参考借鉴价值 ,需要的朋友可以参考下2019-07-07
这篇文章主要给大家介绍了关于如何使用PyTorch优化一个边缘检测器的相关资料,文中通过代码介绍的非常详细,对大家的学习或者工作具有一定的参考借鉴价值,需要的朋友可以参考下2024-09-09
这篇文章主要介绍了python实现的dnspod自动更新dns解析的方法,需要的朋友可以参考下2014-02-02
这篇文章主要介绍了Python简单过滤字母和数字的方法,涉及Python基于内置函数与正则表达式进行字母和数字过滤的相关操作技巧,需要的朋友可以参考下2019-01-01
最新评论