博客

创建一个连续的扁平化的NumPy数组可以使用numpy中的ravel函数或者flatten函数。

使用ravel函数创建扁平化数组

1. 准备数据

首先需要准备一个多维数组作为输入数据，如下所示：

import numpy as np

# 定义一个 2x3 的多维数组
arr = np.array([[1, 2, 3], [4, 5, 6]])

2. 使用ravel函数创建扁平化数组

使用ravel函数将多维数组扁平化，代码如下所示：

# 使用ravel函数创建扁平化数组
flatten_arr = np.ravel(arr)

此时，flatten_arr即为创建的扁平化数组，代码如下所示：

# 输出扁平化数组
print(flatten_arr)

输出结果为：

[1 2 3 4 5 6]

使用flatten函数创建扁平化数组

1. 准备数据

同样需要准备一个多维数组作为输入数据，如下所示：

import numpy as np

# 定义一个 2x3 的多维数组
arr2 = np.array([[7, 8, 9], [10, 11, 12]])

2. 使用flatten函数创建扁平化数组

使用flatten函数将多维数组扁平化，代码如下所示：

# 使用flatten函数创建扁平化数组
flatten_arr2 = arr2.flatten()

此时，flatten_arr2即为创建的扁平化数组，代码如下所示：

# 输出扁平化数组
print(flatten_arr2)

输出结果为：

[ 7  8  9 10 11 12]

通过以上两个示例可以看出，ravel函数和flatten函数都可以将多维数组扁平化为一维数组，使用的方法都很简单。需要注意的是，ravel函数在某些情况下会返回原始数组的视图，而flatten函数始终返回一个拷贝。