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