附加具有相同形状的ndarrays

Question

我有这两个对象：

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

我希望将a到b的元素连接起来，用两个数组中的所有元素创建一个数组。

我该怎么办？

由于

Answer 1

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

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)