创建一个连续的扁平化的NumPy数组

  • Post category:Python

创建一个连续的扁平化的NumPy数组,可以按照以下步骤进行:

步骤一:导入NumPy库

使用以下代码导入NumPy库:

import numpy as np

步骤二:定义数组

创建一个连续的扁平化的NumPy数组,需要先定义一个多维数组,可以使用以下代码定义一个2维数组:

array_2d = np.array([[1, 2], [3, 4]])

上述代码中,我们使用 np.array 函数创建了一个包含两行两列的数组 array_2d,其中第一行为 [1, 2],第二行为 [3, 4]

步骤三:展开数组

使用 np.ravel 函数将多维数组展开成一维数组,可以使用以下代码将 array_2d 二维数组展开成一维数组:

array_1d = np.ravel(array_2d)

上述代码中,我们使用 np.ravel 函数将二维数组 array_2d 展开成一维数组 array_1d

步骤四:验证结果

使用以下代码打印展开后的一维数组,以验证结果是否正确:

print(array_1d)

上述代码中,我们使用 print 函数打印一维数组 array_1d

以下是完整代码示例:

import numpy as np

array_2d = np.array([[1, 2], [3, 4]])
array_1d = np.ravel(array_2d)

print(array_1d)

运行上述代码后,输出结果为:

[1 2 3 4]

另外一种创建连续扁平化的NumPy数组的方式是使用 np.flatten 函数,与使用 np.ravel 函数不同的是,np.flatten 函数会返回一个复制后的数组。示例如下:

import numpy as np

array_2d = np.array([[1, 2], [3, 4]])
array_1d = array_2d.flatten()

print(array_1d)

输出结果与之前示例相同:

[1 2 3 4]