Python中的squeeze方法详解

numpy.squeeze(a, axis=None)

import numpy as np

# 创建一个维度为1的数组
a = np.array([[[1], [2], [3]]])
print("原始数组：")
print(a)
print("原始数组的形状：")
print(a.shape)

# 使用squeeze方法去除数组中的维度为1的轴
b = np.squeeze(a)
print("去除维度为1的轴后的数组：")
print(b)
print("去除维度为1的轴后数组的形状：")
print(b.shape)

原始数组：
[[[1]
  [2]
  [3]]]
原始数组的形状：
(1, 3, 1)
去除维度为1的轴后的数组：
[1 2 3]
去除维度为1的轴后数组的形状：
(3,)

import numpy as np

# 创建一个维度为1的数组
a = np.array([[[1], [2], [3]]])
print("原始数组：")
print(a)
print("原始数组的形状：")
print(a.shape)

# 使用squeeze方法去除特定轴上的维度为1的情况
b = np.squeeze(a, axis=2)
print("去除维度为1的轴后的数组：")
print(b)
print("去除维度为1的轴后数组的形状：")
print(b.shape)

原始数组：
[[[1]
  [2]
  [3]]]
原始数组的形状：
(1, 3, 1)
去除维度为1的轴后的数组：
[[1 2 3]]
去除维度为1的轴后数组的形状：
(1, 3)