Объединение списка двумерных массивов - PullRequest
0 голосов
/ 07 июня 2019

Как мне объединить N, двумерные массивы (размером R x C), чтобы создать трехмерный массив фигуры (N, R, C)? Прямо сейчас N-2D массив массивов содержится внутри списка, и я хочу, чтобы он стал трехмерным массивом. Скажем, X - это мой список двумерных массивов, если я просто сделаю np.array(X), я получу что-то по форме (N,). Если я сделаю np.vstack(X), я получу что-то по форме (N x R, C). Как мне решить эту проблему?

Ответы [ 2 ]

0 голосов
/ 07 июня 2019

вы можете просто использовать:

np.array([np.array(x) for x in ArrayList])
0 голосов
/ 07 июня 2019

Вы можете использовать np.stack :

test = np.stack([np.ones([2, 3]) for _ in range(4)]) 
print(test.shape) # (4, 2, 3)
...