python - 在 numpy 中堆叠数组

Question

我有两个数组:

A = np.array([1, 2, 3])
B = np.array([2, 3, 4])
C = np.stack((A, B), axis=0)

print C.shape
(2, 3)

形状不应该是 (6,) 吗？

Answer 1

使用 np.stack() 函数，您可以指定要将哪个 轴 视为索引轴。因此，如您所见，您永远不会得到 6 的形状，对于此示例，只有 (2,3) 或 (3,2)取决于您选择的轴。

见下文:

A = np.array([1, 2, 3])
B = np.array([2, 3, 4])
arrays = [A, B]

使用这段代码:

print(np.stack(arrays, axis=0))

你得到这个输出:

[[1 2 3]
 [2 3 4]]

使用此代码:

print(np.stack(arrays, axis=1))

你得到这个输出:

[[1 2]
 [2 3]
 [3 4]]