np.swapaxes
时间: 2023-10-30 10:03:15 浏览: 48
`np.swapaxes` 是 NumPy 库中一个函数,用于交换数组的两个轴(维度)。
它的语法如下:
```python
np.swapaxes(arr, axis1, axis2)
```
其中,`arr` 是要进行轴交换的数组,`axis1` 和 `axis2` 是要交换的两个轴的编号。
例如,假设我们有一个形状为 (2, 3, 4) 的三维数组 `arr`:
```python
import numpy as np
arr = np.arange(24).reshape((2, 3,4))
print(arr)
```
输出:
```
array([[[ 0, 1, 2, 3],
[ 4, 5, 6, 7],
[ 8, 9, 10, 11]],
[[12, 13, 14, 15],
[16, 17, 18, 19],
[20, 21, 22, 23]]])
```
若我们想交换第一个轴和第二个轴,可以使用 `np.swapaxes`:
```python
arr_swapped = np.swapaxes(arr, 0, 1)
print(arr_swapped)
```
输出:
```
array([[[ 0, 1, 2, 3],
[12, 13, 14, 15]],
[[ 4, 5, 6, 7],
[16, 17, 18, 19]],
[[ 8, 9, 10, 11],
[20, 21, 22, 23]]])
```
可以看到,原先的第一个轴变成了第二个轴,第二个轴变成了第一个轴,同时数组的形状也发生了相应的变化。
相关推荐
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)