创建连续的扁平化的NumPy数组可以通过NumPy库中的reshape函数和arange函数来实现,步骤如下:
1.导入NumPy库。
import numpy as np
2.使用arange函数创建一个具有连续数字的一维数组。
nums = np.arange(12)
3.使用reshape函数将一维数组转换成二维数组(N行M列)。
arr = nums.reshape((3, 4))
其中(3,4)表示数组有3行4列,也就是一共有12个元素,等于原来一维数组的长度。
4.将二维数组扁平化转成一维数组。
flat_arr = arr.flatten()
示例1:
import numpy as np
#创建一维数组
nums = np.arange(12)
#转换成二维数组
arr = nums.reshape((3, 4))
#将二维数组扁平化
flat_arr = arr.flatten()
print(flat_arr)
输出结果:
[ 0 1 2 3 4 5 6 7 8 9 10 11]
示例2:
import numpy as np
#创建一维数组
nums = np.arange(16)
#转换成三维数组
arr = nums.reshape((2, 2, 4))
#将三维数组扁平化
flat_arr = arr.flatten()
print(flat_arr)
输出结果:
[ 0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15]
以上步骤,就可以创建出一个连续的扁平化的NumPy数组了。
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:创建一个连续的扁平化的NumPy数组 - Python技术站