gpt4 book ai didi

详解Numpy扩充矩阵维度(np.expand_dims, np.newaxis)和删除维度(np.squeeze)的方法

转载 作者:qq735679552 更新时间:2022-09-29 22:32:09 38 4
gpt4 key购买 nike

CFSDN坚持开源创造价值,我们致力于搭建一个资源共享平台,让每一个IT人在这里找到属于你的精彩世界.

这篇CFSDN的博客文章详解Numpy扩充矩阵维度(np.expand_dims, np.newaxis)和删除维度(np.squeeze)的方法由作者收集整理,如果你对这篇文章有兴趣,记得点赞哟.

在操作矩阵的时候,不同的接口对于矩阵的输入维度要求不同,输入可能为1-D,2-D,3-D等等。下面介绍一下使用Numpy进行矩阵维度变更的相关方法。主要包括以下几种:

1、np.newaxis扩充矩阵维度 。

2、np.expand_dims扩充矩阵维度 。

3、np.squeeze删除矩阵中维度大小为1的维度 。

np.newaxis,np.expand_dims扩充矩阵维度:

?
1
2
3
4
5
6
7
8
9
10
11
12
import numpy as np
 
x = np.arange( 8 ).reshape( 2 , 4 )
print (x.shape)
 
# 添加第0维,输出shape -> (1, 2, 4)
x1 = x[np.newaxis, :]
print (x1.shape)
 
# 添加第1维, 输出shape -> (2, 1, 4)
x2 = np.expand_dims(x, axis = 1 )
print (x2.shape)

输出结果:

(2, 4) (1, 2, 4) (2, 1, 4) 。

np.squeeze降低矩阵维度:

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
"""
  squeeze 函数:从数组的形状中删除单维度条目,即把shape中为1的维度去掉
  用法:numpy.squeeze(a,axis = None)
   1)a表示输入的数组;
   2)axis用于指定需要删除的维度,但是指定的维度必须为单维度,否则将会报错;
   3)axis的取值可为None 或 int 或 tuple of ints, 可选。若axis为空,则删除所有单维度的条目;
   4)返回值:数组
   5) 不会修改原数组;
"""
import numpy as np
print ( "#" * 40 , "原始数据" , "#" * 40 )
x = np.arange( 10 ).reshape( 1 , 1 , 10 , 1 )
print (x.shape)
print (x)
 
print ( "#" * 40 , "去掉axis=0这个维度" , "#" * 40 )
x_squeeze_0 = np.squeeze(x, axis = 0 )
print (x_squeeze_0.shape, x_squeeze_0)
 
print ( "#" * 40 , "去掉axis=3这个维度" , "#" * 40 )
x_squeeze_3 = np.squeeze(x, axis = 3 )
print (x_squeeze_3.shape, x_squeeze_3)
 
print ( "#" * 40 , "去掉axis=0, axis=1这两个维度" , "#" * 40 )
x_squeeze_0_1 = np.squeeze(x, axis = ( 0 , 1 ))
print (x_squeeze_0_1.shape, x_squeeze_0_1)
 
print ( "#" * 40 , "去掉所有1维的维度" , "#" * 40 )
x_squeeze = np.squeeze(x)
print (x_squeeze.shape, x_squeeze)
 
print ( "#" * 40 , "去掉不是1维的维度,抛异常" , "#" * 40 )
try :
  x_squeeze = np.squeeze(x, axis = 2 )
  print (x_squeeze.shape, x_squeeze)
except Exception as e:
  print (e)

输出结果:

######################################## 原始数据 ######################################## (1, 1, 10, 1) [[[[0]    [1]    [2]    [3]    [4]    [5]    [6]    [7]    [8]    [9]]]] ######################################## 去掉axis=0这个维度 ######################################## (1, 10, 1) [[[0]   [1]   [2]   [3]   [4]   [5]   [6]   [7]   [8]   [9]]] ######################################## 去掉axis=3这个维度 ######################################## (1, 1, 10) [[[0 1 2 3 4 5 6 7 8 9]]] ######################################## 去掉axis=0, axis=1这两个维度 ######################################## (10, 1) [[0]  [1]  [2]  [3]  [4]  [5]  [6]  [7]  [8]  [9]] ######################################## 去掉所有1维的维度 ######################################## (10,) [0 1 2 3 4 5 6 7 8 9] ######################################## 去掉不是1维的维度,抛异常 ######################################## cannot select an axis to squeeze out which has size not equal to one 。

参考链接 。

到此这篇关于详解Numpy扩充矩阵维度(np.expand_dims, np.newaxis)和删除维度(np.squeeze)的方法的文章就介绍到这了,更多相关Numpy扩充矩阵维度和删除维度内容请搜索我以前的文章或继续浏览下面的相关文章希望大家以后多多支持我! 。

原文链接:https://blog.csdn.net/cxx654/article/details/105762021/ 。

最后此篇关于详解Numpy扩充矩阵维度(np.expand_dims, np.newaxis)和删除维度(np.squeeze)的方法的文章就讲到这里了,如果你想了解更多关于详解Numpy扩充矩阵维度(np.expand_dims, np.newaxis)和删除维度(np.squeeze)的方法的内容请搜索CFSDN的文章或继续浏览相关文章,希望大家以后支持我的博客! 。

38 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com