多次元ndarray(n=4)にスライス(n=3)を追加したい。

Question

(4,25,3)といったshapeのndarrayがあるとして、同じ次元同士をnp.stackで組み合わせて(2,4,25,3) といったスタックにすることはできたのですが、これに更に(4, 25, 3)の行列をスライスとして先の(2,4,25,3)の行列にくっつけて（3,4,25,3）にするにはどのようにしたらよいのでしょうか？

Answer 1

こんな感じでどうでしょうか。

import numpy as np

# 入力

a = np.ones((2, 4, 25, 3))
b = np.ones((4, 25, 3)) * 2

# 処理

b2 = np.expand_dims(b, axis=0)
print(b2.shape)  # -> (1, 4, 25, 3)

c = np.concatenate([a, b2], axis=0)
print(c.shape)  # -> (3, 4, 25, 3)

Comments

[chgrios]

早急な解答ありがとうございました。なるほど、そういった方法があるのですね、知りませんでした。どうして３行目でｂの行列を二倍しているのか教えていただけませんか？

[mokemokechicken]

あ、結果を確認するために print(c) をしたときに、どこに追加されたかわかりやすいので2倍しています。
紛らわしかったですね...

[chgrios]

ありがとうございました！スッキリしました！