使用 Numpy 将多维数组相互附加/插入

Question

import pandas as pd
import numpy as np

x1 = np.random.randint(0,2000,(12,220,80))
x2 = np.random.randint(0,2000,(12,220,1000))

我目前有两个 3-D 数组，我想将它们组合在一起以创建一个 4-D 数组并寻找最有效的方法

我想将它们组合起来，使它们具有 (12,220,81,1000) 形状，以便 x1 重复 1000 次，将第二个数组的每个元素附加到第一个数组的末尾。 我已经尝试了沿各个轴的np.insert 、 np.concatenate和np.append不同组合，但似乎无法让它产生所需的形状

感谢您提前提供任何帮助

Answer 1

使x1 a (12,220,80,1) 并在最后一个轴上repeat以获得 (12,220,80,1000)。 同样将x2扩展到 (12,200,1,1000)。 然后你可以在axis=2上连接。

Answer 2

在@hpaulj 的回应之后起作用的解决方案。 它以每个循环 2.35 s ± 109 ms 的速度执行。 如果有人更快地意识到任何事情，那将是惊人的，但这很有效

x1_ = np.repeat(x1[:,:,:,None],np.shape(x2)[2],axis= -1)
x2_ = np.repeat(x2[:,:,None,:],1,axis = 2)
final = np.concatenate((x1_,x2_),axis = 2)