二次元配列の次元を変換したいです。

Question

arr1=[[1 2 3][4 5 6][7 8 9].....[n-2 n-1 n]]

上記のようなnumpy配列を以下のような配列に変換したいです。

arr2=[[[1 2 3][4 5 6]][[7 8 9][10 11 12]][.....[n-2 n-1 n]]]

arr1では[1 2 3],[4 5 6]...が大きな二次元配列に収まっているのですが、arr2は[1 2 3],[4 5 6]と二つの要素の段階で[[1 2 3][4 5 6]]と二次元配列になる形となっております。

このように、ある特定（上の例では2つ）の配列の段階で次元を増やし新たな配列を作り出すことは可能でしょうか。

稚拙な説明文ですが、ご教授いただければ幸いです。

Answer 1

以下のような感じでしょうか。

import numpy as np

# テストデータ
a = np.arange(1,3*2*4+1).reshape(-1,3)
print(a)
"""
[[ 1  2  3]
 [ 4  5  6]
 [ 7  8  9]
 [10 11 12]
 [13 14 15]
 [16 17 18]
 [19 20 21]
 [22 23 24]]
"""

b = a.reshape(-1,2,3)
print(b)
"""
[[[ 1  2  3]
  [ 4  5  6]]

 [[ 7  8  9]
  [10 11 12]]

 [[13 14 15]
  [16 17 18]]

 [[19 20 21]
  [22 23 24]]]
"""

Comments

[secondson]

迅速なご回答ありがとうございました。