Python用不同大小的数组连接起来

Question

我有一个文件夹，里面有几个不同的变量名，还有几个不同的描述符。例如：

 A1, B1, C1, A2, B2, C2, A3, B3, C3, ...

这些文件的numpy形状为[#，5，5，1]，其中每个变量和数字的#将不同。但是，我想创建一个主数组，包含所有a、B和C，它们已经连接在一起。你知道吗

例如：

 A1.shape = [1426,5,5,1]
 A2.shape = [1322,5,5,1]
 A3.shape = [1112,5,5,1]
 1426 + 1322 + 1112 = 3860
 allA.shape = [3860,5,5,1]

我尝试了两种不同的方法，包括将'allA'预分配为空numpy数组和连接，但错误是“除了连接轴以外的所有输入数组维度都必须完全匹配”。正确的，甚至更简单的方法是什么？你知道吗

Answer 1

你知道吗numpy.concatenate连接在这种情况下应该能正常工作。你知道吗

import numpy as np

a = np.zeros((10, 5, 5, 1))
b = np.zeros((15, 5, 5, 1))
c = np.zeros((20, 5, 5, 1))

res = np.concatenate((a, b, c), axis=0)  # concatenation

print(res.shape)  # (45, 5, 5, 1)