Python - 将多个具有相同维度的二维数组合并为一个二维数组

Question

我有 3 个具有相同尺寸的 2D numpy 浮点数组（R、G、B）。我想将它合并到一个二维数组中，使新二维数组中的每个元素都是一个由 3 个浮点数组成的数组。以下是我的初始代码：

image = cv2.imread('hurricane katrina 1.jpg', cv2.IMREAD_GRAYSCALE)

a = 255
b = 2 * (np.pi/255)
c = np.pi / 5


R = a * np.absolute(np.sin(b * image))
G = a * np.absolute(np.sin(b * image + c))
B = a * np.absolute(np.sin(b * image + 2 * c))

如何在不遍历数组的情况下在 Python 中执行此操作？谢谢。

更新：

我想合并 3 个数组。比如说

R[0][0] = 3 
G[0][0] = 4 
B[0][0] = 6

那么合并数组就是

RGB[0][0] = (3,4,6)

这对于数组中的所有元素都是正确的。

所以最终的输出形状将是(1000,775,3)

Answer 1

np.concatenate((a, b,c), axis=see what you want, probably 1)

或者 你也可以使用 hstack...（它们都是同一个东西...）

Doing, rgb_transform = np.concatenate(R,G,B), throws an error. TypeError: only integer scalar arrays can be converted to a scalar index, 但它不应该抛出，因为我已经在括号中传递了它们

为了解决这个问题，可能 做这个 np.concatenate([a, b, c], axis=see what you want, probably 1)

请参考docs

编辑

问的和上面的有点不同，所以正确的答案是做 np.stack((r,g,b),2) 因为最终输出是 (1000,775,3)...