2024年1月7日发(作者:)
numpy swapaxes用法
es()函数是对数组的维度进行重组的函数,它可以交换数组的任意两个维度位置。
函数的用法如下:
es(arr, axis1, axis2)
参数说明:
arr:需要重组维度的数组
axis1:需要交换的第一个维度位置
返回值:
返回交换后的数组
简单的例子可以帮助你更好地理解swapaxes函数的用途:
import numpy as np
a = ([[[1,2],[3,4]],[[5,6],[7,8]]])
print ('原数组:')
print (a)
print ('n')
#现在交换轴0(深度方向)到轴2(宽度方向)
print ('调用 swapaxes 函数后的数组:')
print (es(a,2,0))
输出:
原数组:
[[[1 2]
[3 4]]
[[5 6]
[7 8]]]
从输出结果可以看到,原数组a有三个维度,swapaxes函数把位置2(宽度方向)与位置0(深度方向)进行了交换,重组后数组的第一个维度变成了原来的第二个维度,第二个维度变成了原来的第三个维度,第三个维度变成了原来的第一个维度。
除此之外,swapaxes函数还有其他很实用的用途,比如:
1.将一维数组转换为行向量或列向量
调用 swapaxes 函数后的数组:
[[0]
[1]
[2]
[3]
[4]
[5]
[6]
[7]]
从输出结果可以看到,swapaxes函数把位置0(原来的行方向)与位置1(原来的列方向)交换,重组结果就是一个列向量。
2.数组转置
在使用swapaxes函数时,需要特别注意数组的维度顺序。如果顺序不对,可能会得到意想不到的结果。因此,在处理多维数组时,我们通常需要手动打印每个维度的形状,以确保swapaxes函数的正确使用。
发布者:admin,转转请注明出处:http://www.yc00.com/news/1704590172a1359290.html
评论列表(0条)