python - 将 Numpy 多维数组添加在一起

Question

如果我有两个 Numpy 数组:形状均为 (50,5,5)，我如何将它们相加以获得形状为 (50,5,10) 的数组？

Answer 1

使用concatenate :

import numpy as np
n = 50 * 5 * 5
a = np.random.random(size=n).reshape(50,5,5)
b = np.random.random(size=n).reshape(50,5,5)

np.concatenate([a,b], axis=2).shape # (50, 5, 10)