在numpy中,可以使用numpy.ndenumerate()
函数来枚举2D数组,并将枚举结果追加到新数组中。
具体步骤如下:
import numpy as np
arr = np.array([[1, 2], [3, 4], [5, 6]])
new_arr = np.empty((0, 3), int)
numpy.ndenumerate()
函数来枚举2D数组,并将枚举结果追加到新数组中:for index, value in np.ndenumerate(arr):
row = np.array([index[0], index[1], value])
new_arr = np.append(new_arr, [row], axis=0)
在上述代码中,index
表示当前元素的索引,value
表示当前元素的值。我们将索引和值组成一个新的一维数组row
,然后将row
追加到新数组new_arr
中。
最后,可以打印新数组new_arr
来查看结果:print(new_arr)
完整代码示例:
import numpy as np
arr = np.array([[1, 2], [3, 4], [5, 6]])
new_arr = np.empty((0, 3), int)
for index, value in np.ndenumerate(arr):
row = np.array([index[0], index[1], value])
new_arr = np.append(new_arr, [row], axis=0)
print(new_arr)
这样,就可以在numpy中枚举2D数组并追加到新数组中了。
推荐的腾讯云相关产品:腾讯云云服务器(CVM)和腾讯云对象存储(COS)。
领取专属 10元无门槛券
手把手带您无忧上云