numpy中的转置Transpose和.T以及轴对换swapaxis
作者:互联网
numpy中Transpose和.T以及swapaxis都能对ndarray在不同维度上进行转置,交换操作,下面分别介绍这三个函数的用法,相互对比以加深影响。
1.Transpose
这个函数如果括号内不带参数,和.T效果一样,见下面例子:
import numpy as np
arr=np.arange(16).reshape(2,2,4)
print('arr:')
print(arr)
print('arr.transpose:')
print(arr.transpose())
print('arr.transpose(2,1,0):')
print(arr.transpose(2,1,0))
print('arr.T:')
print(arr.T)
结果:
arr:
[[[ 0 1 2 3]
[ 4 5 6 7]]
[[ 8 9 10 11]
[12 13 14 15]]]
arr.transpose:
[[[ 0 8]
[ 4 12]]
[[ 1 9]
[ 5 13]]
[[ 2 10]
[ 6 14]]
[[ 3 11]
[ 7 15]]]
arr.transpose(2,1,0):
[[[ 0 8]
[ 4 12]]
[[ 1 9]
[ 5 13]]
[[ 2 10]
[ 6 14]]
[[ 3 11]
[ 7 15]]]
arr.T:
[[[ 0 8]
[ 4 12]]
[[ 1 9]
[ 5 13]]
[[ 2 10]
[ 6 14]]
[[ 3 11]
[ 7 15]]]
从上面的例子中可以看出arr.transpose(), arr.T ,arr.transpose(2,1,0)的结果是相同的。现在我们详细看看transpose()带参数的情况,就会明白为什么arr.transpose(2,1,0)有这样的结果。
首先因为arr这个数组有三个维度,三个维度的编号对应为(0,1,2),假如我们需要拿到arr的7这个元素,肯定需要些三个维度的值,7的第一个维度为0,第二个维度为1,第三个3,所以arr[0,1,3] = 7。
再回到transpose()这个函数,它里面参数的顺序就是维度的顺序,比如transpose(2,1,0),就是把之前第三个维度转为第一个维度,之前的第二个维度不变,之前的第一个维度变为第三个维度,我们继续拿7这个值来说,之前的索引为[0,1,3],按照我们的转换方法,把之前的第三维度变为第一维度,之前的第一维度变为第三维度,那么现在7的索引就是(3,1,0)
同理所有的数组内的数字都是这样变得,这就是transpose()内参数的变化。
2.swapaxis
swapaxes接受一对轴编号,或者叫一对维度编号,里面只能是两个参数,且不分顺序。
import numpy as np
arr=np.arange(16).reshape(2,2,4)
print('arr:')
print(arr)
print('arr.swapaxes(1,2):')
print(arr.swapaxes(1,2))
#print('arr.swapaxes(2,1):')
#print(arr.swapaxes(2,1))
print('arr.transpose(0,2,1):')
print(arr.transpose(0,2,1))
结果:
arr:
[[[ 0 1 2 3]
[ 4 5 6 7]]
[[ 8 9 10 11]
[12 13 14 15]]]
arr.swapaxes(1,2):
[[[ 0 4]
[ 1 5]
[ 2 6]
[ 3 7]]
[[ 8 12]
[ 9 13]
[10 14]
[11 15]]]
arr.swapaxes(2,1):
[[[ 0 4]
[ 1 5]
[ 2 6]
[ 3 7]]
[[ 8 12]
[ 9 13]
[10 14]
[11 15]]]
arr.transpose(0,2,1):
[[[ 0 4]
[ 1 5]
[ 2 6]
[ 3 7]]
[[ 8 12]
[ 9 13]
[10 14]
[11 15]]]
从上面的例子可以看出arr.swapaxes(1,2),arr.swapaxes(2,1),arr.transpose(0,2,1)效果是等价的。
参考文章:
https://blog.csdn.net/qq1483661204/article/details/70543952
标签:11,arr,转置,Transpose,对换,transpose,维度,swapaxes,print 来源: https://blog.csdn.net/u014157109/article/details/89787623