Python 乱数行列

Question

前提・実現したいこと

　2つの3x4の乱数行列を列方向に追加し3x2x4の3次元の行列を作成したいです。

発生している問題・エラーメッセージ

　2つの行列を作り、4×2×3の3次元行列は作れるのですが、3x2x4はどうすればいいのでしょうか

該当のソースコード

import numpy as np

array1 = np.arange(12).reshape(3,4)
array2 = np.arange(13,25).reshape(3,4)

print(np.dstack([array1,array2]))

Comments

[Yasumichi]

一応、作りたい行列のイメージを追記してもらえますか？

Answer 1

np.stack()を使うと良いのではないでしょうか

https://docs.scipy.org/doc/numpy/reference/generated/numpy.stack.html

import numpy as np

array1 = np.arange(12).reshape(3,4)
array2 = np.arange(13,25).reshape(3,4)

ret = np.stack([array1,array2], axis=1)
print(ret)
#[[[ 0  1  2  3]
#  [13 14 15 16]]
#
# [[ 4  5  6  7]
#  [17 18 19 20]]
#
# [[ 8  9 10 11]
#  [21 22 23 24]]]

print(ret.shape)
#(3, 2, 4)