回答編集履歴
1
list対応やめ
test
CHANGED
@@ -6,13 +6,11 @@
|
|
6
6
|
|
7
7
|
def argmins(x):
|
8
8
|
|
9
|
-
return [v[0] for v in enumerate(x) if v[1] == min(
|
9
|
+
return [v[0] for v in enumerate(x) if v[1] == x.min()]
|
10
10
|
|
11
11
|
|
12
12
|
|
13
13
|
l = [3, 1, 2, 3, 1, 1]
|
14
|
-
|
15
|
-
print(argmins(l)) #=> [1, 4, 5]
|
16
14
|
|
17
15
|
print(argmins(np.array(l))) #=> [1, 4, 5]
|
18
16
|
|