我觉得我缺少一些文档,但我找不到关于这个特定示例的任何内容 - 一切都只是关于连接或堆叠数组。
我有形状为 (2,3) 的数组 x 和数组 y
x = [[1,2,3],[4,5,6]]
y = [[7,8,9],[10,11,12]]
x = 1 2 3
4 5 6
y = 7 8 9
10 11 12
我想要形状为 (2,3,2) 的数组 z,如下所示
z = [[[1,7],[2,8],[3,9]],[[4,10],[5,11],[6,12]]]
z = [1,7] [2,8] [3,9]
[4 10] [5 11] [6 12]
基本上在每个位置连接 x 和 y 元素。
请您参考如下方法:
听起来您正在寻找的函数是 stack(),使用它沿 3 维堆叠。
import numpy as np
x = np.asarray([[1,2,3],[4,5,6]])
y = np.asarray([[7,8,9],[10,11,12]])
z = np.stack((x, y), 2)
