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