回答編集履歴
1
結果例の追加
test
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
numpyを使えば簡単に実装できると思います。
|
|
2
2
|
|
|
3
|
-
```
|
|
3
|
+
```python
|
|
4
4
|
|
|
5
5
|
import numpy as np
|
|
6
6
|
|
|
@@ -41,3 +41,21 @@
|
|
|
41
41
|
np.where(res > 3000,1, 0)
|
|
42
42
|
|
|
43
43
|
```
|
|
44
|
+
|
|
45
|
+
|
|
46
|
+
|
|
47
|
+
出力例
|
|
48
|
+
|
|
49
|
+
```
|
|
50
|
+
|
|
51
|
+
array([[0, 0, 0, 0, 0],
|
|
52
|
+
|
|
53
|
+
[1, 0, 0, 0, 1],
|
|
54
|
+
|
|
55
|
+
[0, 0, 0, 0, 0],
|
|
56
|
+
|
|
57
|
+
[1, 0, 0, 0, 1],
|
|
58
|
+
|
|
59
|
+
[0, 0, 0, 0, 0]])
|
|
60
|
+
|
|
61
|
+
```
|