squeeze函数
时间: 2023-08-11 14:21:51 浏览: 67
`squeeze()`函数是一个NumPy中的函数,用于从数组的形状中删除单维度条目。也就是说,如果数组的某个维度的长度为1,那么该维度可以被删除,这样数组的形状会发生变化。
`squeeze()`函数的语法如下:
```
numpy.squeeze(arr, axis=None)
```
其中,arr表示要进行操作的输入数组,axis是一个可选的参数,表示要删除的维度。如果未指定axis,则删除所有长度为1的维度。
例如,对于一个形状为(1,3,1)的数组,使用`squeeze()`函数可以将其转换为形状为(3,)的数组,代码如下:
```python
import numpy as np
a = np.array([[[1], [2], [3]]])
b = np.squeeze(a)
print(b.shape) # 输出 (3,)
```
在这个例子中,输入数组a的形状为(1,3,1),其中第一个维度长度为1,因此可以使用`squeeze()`函数将其删除,得到形状为(3,)的输出数组b。
相关推荐
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![](https://img-home.csdnimg.cn/images/20210720083646.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![application/x-zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![docx](https://img-home.csdnimg.cn/images/20210720083331.png)
![docx](https://img-home.csdnimg.cn/images/20210720083331.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![docx](https://img-home.csdnimg.cn/images/20210720083331.png)
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)