在Python中将2d numpy数组展平为1d数组
2dnumpy数组是数组的数组。在本文中，我们将看到如何展平它以使元素成为一维数组。
与扁平化
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]