numpy的squeeze函数使用方法

  • Post category:Python

以下是关于“numpy的squeeze函数使用方法”的完整攻略。

numpy的squeeze函数简介

在NumPy中,squeeze()函数用于从数组的形状中删除单维度条目。例如如果数组a的形状为(1, 3, 1, 5),则使用squeeze()函数可以将其形状变为(3, 5)。

numpy的squeeze函数使用方法

下面是squeeze()函数的使用方法:

numpy.squeeze(arr, axis=None)

其中,arr表示要操作的数组,axis表示要删除的单维度条目的轴。如果axis为None,则删除所有单维度条目。

下面是一个使用squeeze()函数的示例代码:

import numpy as np

# 创建一个形状为(1, 3, 1, 5)的数组
a = np.random.rand(1, 3, 1, 5)

# 使用squeeze()函数删除维度条目
b = np.squeeze(a)

# 输出结果
print('Original shape:', a.shape)
print('Squeezed shape:', b.shape)

在上面的示例代码中,我们首先使用np.random.rand()函数创建了一个形状为(1, 3, 1, 5)的数组a。然后,我们使用np.squeeze()函数删除了a中的单维度条目,并将结果存储在变量b中。最后,我们输出了原始数组和删除单维度条目后的数组。

下面是另一个使用squeeze()函数的示例代码:

import numpy as np

# 创建一个形状为(2, 1, 3, 1, 5)的数组
a = np.random.rand(2, 1, 3, 1, 5)

# 使用squeeze()函数删除第二个轴上的单维度条目
b = np.squeeze(a, axis=1)

# 输出结果
print('Original shape:', a.shape)
print('Squeezed shape:', b.shape)

在上面的示例代码中,我们首先使用np.random.rand()函数创建了一个形状为(2, 1, 3, 1, 5)的数组a。然后,我们使用np.squeeze()函数删除了数组a中第二个轴上的单维度条目,并将结果存储在变量b中。最后,我们输出了原始数组和删除单维度条目后的数组。

综上所述,“numpy的squeeze函数使用方法”的完整攻略包括了squeeze()函数的简介、使用方法和两个示例代码的演示。在实际应用中,可以根据具体的需求选择合适的方法。