演示np.squeeze()的用法
时间: 2023-07-01 15:10:42 浏览: 99
numpy.array 操作使用简单总结
`np.squeeze()`是一个用于从数组中删除单维度条目的函数。下面是一个演示示例:
``` python
import numpy as np
# 创建一个形状为(1, 3, 1, 4)的数组
x = np.array([[[[1, 2, 3, 4]], [[5, 6, 7, 8]], [[9, 10, 11, 12]]]])
# 使用np.squeeze()删除单维度条目
y = np.squeeze(x)
# 打印删除单维度条目前后的数组形状
print("原始数组形状:", x.shape)
print("删除单维度条目后的数组形状:", y.shape)
```
输出结果:
```
原始数组形状: (1, 3, 1, 4)
删除单维度条目后的数组形状: (3, 4)
```
在示例中,我们创建了一个形状为(1, 3, 1, 4)的数组,并使用`np.squeeze()`函数删除了单维度条目。删除单维度条目后,数组的形状从(1, 3, 1, 4)变为了(3, 4)。
阅读全文