3次元配列をNumPyを使い1次元は配列に格納したい

Question

Numpyで定義された深さ5, 縦2 横2で定義された3次元配列xの
深さ方向の値を3個ずつ4個取り出して一括で2次元配列に格納したい。

#3次元配列の定義

x = np.arange(5 * 2 * 2).reshape(5, 2, 2)

#numpyを使い格納した結果がyのようになって欲しい

y = np.array([[ 0,4,8],
[ 1,5,9],
[ 2,6,10],
[ 3,7,11]])

よろしくお願いします。

Comments

[tiitoi]

変換後の配列を見ると、12 ~ 19 までの要素が消えてしまっていますが、それでいいのでしょうか？

[redcatML]

連絡ありがとうございます。
消えてしまって構いません。
深さ方向で3までの値を格納したいです。

Answer 1

以下のようにすればよいです。

1 ravel()[:12]: 一旦1次元配列にし、12個の要素まででスライスする。

[ 0  1  2  3  4  5  6  7  8  9 10 11]

2 reshape(4, 3, order="F"): 列優先順で形状を変更する。

[[ 0  4  8]
 [ 1  5  9]
 [ 2  6 10]
 [ 3  7 11]]

ソースコード

x = np.arange(5 * 2 * 2).reshape(5, 2, 2)
print(x)

y = x.ravel()[:12].reshape(4, 3, order="F")
print(y)
# [[ 0  4  8]
#  [ 1  5  9]
#  [ 2  6 10]
#  [ 3  7 11]]

Comments

[redcatML]

回答ありがとうござます。
スライスしてからreshapeすればよいのですね。
ありがとうございました。