transpose函数用法
时间: 2024-04-25 15:17:03 浏览: 62
transpose函数是numpy库中的一个函数,用于改变数组的轴的顺序。它可以接受一个由轴编号组成的元组作为参数,来改变数组的轴的顺序。下面是一个简单的例子:
```python
import numpy as np
arr = np.array([[1, 2], [3, 4]])
print(np.transpose(arr))
```
输出结果为:
```
[[1 3]
[2 4]]
```
在这个例子中,我们创建了一个2x2的数组,然后使用transpose函数改变了它的轴的顺序,从而得到了一个2x2的转置矩阵。
除了这个简单的例子,transpose函数还可以用于高维数组。在高维数组中,transpose函数可以用于交换任意两个轴。下面是一个例子:
```python
import numpy as np
arr = np.arange(24).reshape((2, 3, 4))
print(arr)
print(np.transpose(arr, (1, 2, 0)))
```
输出结果为:
```
[[[ 0 1 2 3]
[ 4 5 6 7]
[ 8 9 10 11]]
[[12 13 14 15]
[16 17 18 19]
[20 21 22 23]]]
[[[ 0 12]
[ 1 13]
[ 2 14]
[ 3 15]
[ 4 16]
[ 5 17]
[ 6 18]
[ 7 19]
[ 8 20]
[ 9 21]
[10 22]
[11 23]]]
```
在这个例子中,我们创建了一个2x3x4的数组,然后使用transpose函数交换了第一个和第三个轴,从而得到了一个4x2x3的数组。
阅读全文