Skip to main content
 首页 » 编程设计

python之如何合并 2 个 numpy 数组

2025年05月04日14三少

我觉得我缺少一些文档,但我找不到关于这个特定示例的任何内容 - 一切都只是关于连接或堆叠数组。

我有形状为 (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)