回答編集履歴
1
追記
test
CHANGED
@@ -1,3 +1,23 @@
|
|
1
1
|
aとbとで次元が異なりますから、二次元に揃えれば良いです。
|
2
2
|
|
3
3
|
具体的には`b = np.array([[7,8,9]])`という風に。
|
4
|
+
|
5
|
+
|
6
|
+
|
7
|
+
ifで分岐するとしても、次元数で分岐するのが良いでしょう。
|
8
|
+
|
9
|
+
```Python
|
10
|
+
|
11
|
+
def anyfunction(arr):
|
12
|
+
|
13
|
+
if arr.ndim == 1:
|
14
|
+
|
15
|
+
return arr
|
16
|
+
|
17
|
+
elif arr.ndim == 2:
|
18
|
+
|
19
|
+
return arr[-1]
|
20
|
+
|
21
|
+
raise ValueError
|
22
|
+
|
23
|
+
```
|