python中squeeze函数的用法是什么

python

720

2024/4/5 16:18:59

栏目: 编程语言

在Python中，squeeze()函数用于去除数组中的单维度条目。具体来说，它会删除shape为1的维度，即将维度大小为1的维度去掉。

例如，假设有一个形状为(1, 5)的数组，使用squeeze()函数后，它将变成形状为(5,)的数组。

import numpy as np

arr = np.array([[1, 2, 3, 4, 5]])
print(arr.shape)  # 输出 (1, 5)

arr_squeezed = np.squeeze(arr)
print(arr_squeezed.shape)  # 输出 (5,)

squeeze()函数还可以指定去除某个特定的维度，例如axis=0表示去除第一个维度。

import numpy as np

arr = np.array([[[1, 2, 3, 4, 5]]])
print(arr.shape)  # 输出 (1, 1, 5)

arr_squeezed = np.squeeze(arr, axis=0)
print(arr_squeezed.shape)  # 输出 (1, 5)

辰迅云「云服务器」，即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘，价格低至29元/月。点击查看>>

python中squeeze函数的用法是什么

最新知识库

相关标签