博客

改变一个NumPy数组的尺寸通常需要用到NumPy中的reshape方法。下面是使用reshape方法改变NumPy数组尺寸的完整攻略：

1. 导入NumPy包

首先要导入NumPy包，才能使用其中提供的函数和类。

import numpy as np

2. 创建一个NumPy数组

接下来可以创建一个NumPy数组，例如：

a = np.array([[1, 2], [3, 4], [5, 6]])

3. 使用reshape方法改变尺寸

使用reshape方法改变尺寸，通常只需指定新的形状（shape），即想要变成的维度。例如，如果想将一个二维数组转换为一维数组，可以将其形状设为(1, n)或(n,)。

# 将一个二维数组转换为一维数组
a_reshape = a.reshape(6)

如果想将一个一维数组转换为二维数组，可以将其形状设为(行数, 列数)或(行数, -1)，其中-1表示自动计算列数。

# 将一个一维数组转换为二维数组，其中列数自动计算为2
a_reshape = a.reshape(3, -1)

4. 将数组展开为一维数组

有时候需要将一个多维数组展开为一维数组，可以使用numpy.ravel方法。

a_flattened = np.ravel(a)

示例1：从一维数组转换为二维数组

a = np.arange(8)
a_reshape = a.reshape(2, 4)
print(a_reshape)

输出：

array([[0, 1, 2, 3],
       [4, 5, 6, 7]])

示例2：将二维数组展开为一维数组

a = np.array([[1, 2], [3, 4], [5, 6]])
a_flattened = np.ravel(a)
print(a_flattened)

输出：

array([1, 2, 3, 4, 5, 6])