У меня есть папка с несколькими разными именами переменных, а также несколько разных дескрипторов, которые отличаются. Например:
A1, B1, C1, A2, B2, C2, A3, B3, C3, ...
Эти файлы имеют пустую форму [#, 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)