複数のnumpy配列を合体させて一つにしたいです。

Question

前提

2つの多次元numpy配列（ndarray）を合体させて一つにしたいです。

信号処理をしています。
aとbの2種類の信号が違うndarrayとして保存されたものを、一つのndarrayに合成したいです。（a.shapeとb.shapeは同じです)

実現したいこと

例えば、以下のa,b2つのndarrayが与えられた時にcの形で出力をしたいです。

該当のソースコード

a = np.array([[1,3,5], [7,9,11], [13,15,17], [19,21,23]])
b = np.array([[2,4,6], [8,10,12], [14,16,18], [20,22,24]])
（a.shape = b.shape(4,3))
を合体させたら、
↓
c = np.array([  [[1,2],[3,4],[5,6]],  [[7,8],[9,10],[11,12]],  [[13,14],[15,16],[17,18]],  [[19,20],[21,22],[23,24]]  ])
c.shape　>> (4, 3, 2)

試したこと

import numpy as np
a = np.array([[1,3,5],[7,9,11],[13,15,17],[19,21,23]])
b = np.array([[2,4,6],[8,10,12],[14,16,18],[20,22,24]])

c = np.concatenate([a,b], axis=1)
print(c)

[[ 1 3 5 2 4 6]
[ 7 9 11 8 10 12]
[13 15 17 14 16 18]
[19 21 23 20 22 24]]

これをa[0][i]などのfor分で回すのが良いのかと思ったのですが上手くいきません。

補足情報（FW/ツールのバージョンなど）

Python 2.7.12

Answer 1

新しい軸での結合はnumpy.concatenate()ではなく、numpy.stack()を使います。

d = np.stack([a, b], axis=-1)

axisは結合する軸(指定しないと0)で、今回の場合は、axis=2を指定するか、axis=-1 で最後を指定するか、が必要です。

Comments

[shakee]

bsdfan さん

なるほど。これも手元でやりたいことができました！！
大変勉強になります。助かりました。ありがとうございます！

Answer 2

これでどうでしょう？

c = np.array([a.T, b.T]).T

いちおう確かめてみました👉 https://replit.com/@su507/r5vbsrcruvyohr#main.py

Comments

[shakee]

su507さん

ありがとうございます！！やりたかったことはまさにこれでした！転置をすればよかったのですね！勉強になります。ありがとうございます！！！