您可以使用
np.stack它来更简洁地完成此操作：
img = np.array([[1, 2], [3, 4]])stacked_img = np.stack((img,)*3, axis=-1)print(stacked_img) # array([[[1, 1, 1], # [2, 2, 2]], # [[3, 3, 3], # [4, 4, 4]]])
您可以使用
np.stack它来更简洁地完成此操作：
img = np.array([[1, 2], [3, 4]])stacked_img = np.stack((img,)*3, axis=-1)print(stacked_img) # array([[[1, 1, 1], # [2, 2, 2]], # [[3, 3, 3], # [4, 4, 4]]])
名师互学网 版权所有 (c)2021-2022 ICP备案号：晋ICP备2021003244-6号
