numpy增加维度、删除维度的方法

在Numpy中，可以使用reshape()函数增加或删除数组的维度，也可以使用squeeze()函数删除数组中长度为1的维度。下面是详细的讲解和示例：

增加维度

在Numpy中，可以使用reshape()函数增加数组的维度。reshape()函数的用法如下：

import numpy as np

# 创建一个形状为(2, 3)的二维数组
a = np.array([[1, 2, 3], [4, 5, 6]])

# 使用reshape()函数增加一个维度
b = a.reshape((2, 3, 1))

# 打印结果
print(a)
print(b)

在上面的示例中，我们首先使用np.array()函数创建了一个二维数组a，然后使用reshape()函数将其增加了一个维度，并将结果保存在变量b中。最后，使用print()函数打印出了原数组a和增加维度后的b。

需要注意的是，增加维度后数组元素个数须与原数组元素个数相同，否则会报错。

删除维度

在Numpy中，可以使用squeeze()函数删除数组中长度为1的维度。squeeze()函数的用法如下：

import numpy as np

# 创建一个形状为(2, 3, 1)的三维数组
a = np.array([[[1], [2], [3]], [[4 [5], [6]]])

# 使用squeeze()函数删除长度为1的维度
b = np.squeeze(a)

# 打印结果
print(a)
print(b)

在上面的示例中，我们首先使用np.array()函数创建了一个三维数组a，其中第三维度的长度为1。然后使用squeeze()函数删除了长度为1的维度，并将结果保存在变量b中。最后，使用print()打印出了原数组a和删除维度后的b。

需要注意的是，删除维度后数组元素个数必须与原数组元素个数相同，否则会报错。

示例一：增加度

import numpy as np

# 创建一个形状为(2, 3)的二维数组
a = np.array([[1, 2, 3], [4, 5, 6]])

# 使用reshape()函数增加一个维度
b = a.reshape((2, 3, 1))

# 打印结果
print(a)
print(b)

在上面的示例中，我们首先使用np.array()函数创建了一个二维数组a，然使用reshape()函数将其增加了一个维度，并将结果保存在变量b中。最后，使用print()函数打印出了原数组a和增加维度后的b。

示例二：删除维度

import numpy as np

# 创建一个形状为(2, 3, 1)的三维数组
a = np.array([[[1], [2], [3]], [[4], [5], [6]]])

# 使用squeeze()函数删除长度为1的维度
b = np.squeeze(a)

# 打印结果
print(a)
print(b)

在上面的示例中，我们首先使用np.array()函数创建了一个三维数组，其中第三个维度的长度为1。然后使用squeeze()函数删除了长度为1的维度，并将结果保存在变量b中。最后，使用print()函数打印出了原数组a和删除维度后的b。