python - 如何将多个 numpy 2d 数组堆叠到一个 3d 数组中？

Question

这是我的代码:

img = imread("lena.jpg")
for channel in range(3):
    res = filter(img[:,:,channel], filter)
    # todo: stack to 3d here

正如您所看到的，我正在为图片中的每个 channel 应用一些过滤器。如何将它们堆叠回 3d 数组？ (=原始图像形状)

谢谢

Answer 1

您可以使用np.dstack :

import numpy as np

image = np.random.randint(100, size=(100, 100, 3))

r, g, b = image[:, :, 0], image[:, :, 1], image[:, :, 2]

result = np.dstack((r, g, b))

print("image shape", image.shape)
print("result shape", result.shape)

输出

image shape (100, 100, 3)
result shape (100, 100, 3)