Объединить матрицу в NumPy - PullRequest
       29

Объединить матрицу в NumPy

2 голосов
/ 20 декабря 2009

Предположим, у меня есть три "листа" матрицы a, b и c, каждый с одинаковым размером m n p. И я хочу объединить их, чтобы получить новую матрицу m n p * 3, элемент (i, j, k) которой (a [i, j, k], b [i, j, k] , с [I, J, K]). Какую команду мне использовать? Команда dstack здесь не работает. Спасибо.

Ответы [ 2 ]

4 голосов
/ 20 декабря 2009

Еще один лайнер будет:

result = numpy.array( (a,b,c) ).transpose( (1,2,3,0) )

или более информативный метод:

result = empty( (m,n,p,3) )
result[:,:,:,0] = a
result[:,:,:,1] = b
result[:,:,:,2] = c
2 голосов
/ 20 декабря 2009

Я думаю, что вы хотите:

np.concatenate([np.expand_dims(x, -1) for x in (a, b, c)], axis=3)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...