在Python中将2d numpy数组展平为1d数组

2d numpy数组是数组的数组。在本文中，我们将看到如何展平它以使元素成为一维数组。

与扁平化

numpy中的flatten函数是将2d数组转换为1D数组的直接方法。

示例

import numpy as np
array2D = np.array([[31, 12, 43], [21, 9, 16], [0, 9, 0]])
# printing initial arrays
print("Given array:\n",array2D)
# Using flatten()res = array2D.flatten()
# Result
print("Flattened array:\n ", res)

输出结果

运行上面的代码给我们以下结果-

Given array:
[[31 12 43]
[21 9 16]
[ 0 9 0]]
Flattened array:
[31 12 43 21 9 16 0 9 0]

带旅行

还有一个名为ravel的函数，它将完成将2D数组展平为1D的类似操作。

示例

import numpy as np
array2D = np.array([[31, 12, 43], [21, 9, 16], [0, 9, 0]])
# printing initial arrays
print("Given array:\n",array2D)
# Using ravel
res = array2D.ravel()
# Result
print("Flattened array:\n ", res)

输出结果

运行上面的代码给我们以下结果-

Given array:
[[31 12 43]
[21 9 16]
[ 0 9 0]]
Flattened array:
[31 12 43 21 9 16 0 9 0]