回答編集履歴
1
edit
answer
CHANGED
|
@@ -1,1 +1,17 @@
|
|
|
1
|
-
np.vsplitではどうしていけないのでしょうか？
|
|
1
|
+
np.vsplitではどうしていけないのでしょうか？
|
|
2
|
+
|
|
3
|
+
---
|
|
4
|
+
|
|
5
|
+
分割してNからN-1次元の配列を得たいと勘違いしていました。
|
|
6
|
+
|
|
7
|
+
1、Nから2次元配列
|
|
8
|
+
2、2次元配列の行と列を元のN次元配列のi、j軸
|
|
9
|
+
という目標であれば、
|
|
10
|
+
```python
|
|
11
|
+
def to_2d(A, i, j):
|
|
12
|
+
tmp = A.swapaxes(i, -2)
|
|
13
|
+
tmp = tmp.swapaxes(j, -1)
|
|
14
|
+
tmp = tmp.reshape(-1, tmp.shape[-2], tmp.shape[-1])
|
|
15
|
+
return np.vsplit(tmp, tmp.shape[0])
|
|
16
|
+
```
|
|
17
|
+
のような関数を作ればよいはずです。
|