Python使用不同大小的数组连接

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

 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数组和连接，但错误是“除了连接轴之外的所有输入数组维度都必须完全匹配”。什么是正确的，甚至更简单的方法呢？

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)