如何在Python中合並兩個不同維度的arrays？

Question

我有兩個 arrays，分別是 (150,1400,1200) 和 (150,1400)。 我想知道如何在 Python 中以行和列的方式組合這兩個 arrays？

Answer 1

IIUC，您需要向b添加一維然后使用numpy.concatenate

a = np.random.rand(150, 1400, 1200)
print('a.shape:',a.shape)
b = np.random.rand(150, 1400)
print('b.shape:',b.shape)
print('b[..., None].shape:',b[..., None].shape)

c = np.concatenate([a, b[..., None]], axis=2)
print('c.shape:',c.shape)

Output：

a.shape: (150, 1400, 1200)
b.shape: (150, 1400)
b[..., None].shape: (150, 1400, 1)
c.shape: (150, 1400, 1201)

Answer 2

嘿，你可以看看下面的鏈接來解決你的問題

連接兩個不同維度的 arrays numpy