python - 附加具有相同形状的 ndarray

Question

我有这两个对象：

a.shape
(10000, 32, 32, 3)
b.shape
(50000, 32, 32, 3)

我想将 a 的元素连接到 b 以创建一个数组，其中包含两个数组中的所有元素。

我该怎么做？

谢谢

Answer 1

如果需要简单的连接，请使用numpy.concatenate. 请参阅文档。

import numpy as np

a = np.ones((10000, 32, 32, 3))
b = np.ones((50000, 32, 32, 3))

c = np.concatenate((a, b))
print(c.shape)

>>> (60000, 32, 32, 3)