質問編集履歴
1
スクリプトの追記
test
CHANGED
|
File without changes
|
test
CHANGED
|
@@ -25,3 +25,69 @@
|
|
|
25
25
|
### 使用しているバージョン
|
|
26
26
|
|
|
27
27
|
Python 3.8.5
|
|
28
|
+
|
|
29
|
+
|
|
30
|
+
|
|
31
|
+
### 追記
|
|
32
|
+
|
|
33
|
+
@can110さんの方法で以下のスクリプトで試しました。CSVファイルは``3FloorLv4_result_2021-10-20-20-47-55.csv``という名前です。
|
|
34
|
+
|
|
35
|
+
```test.py
|
|
36
|
+
|
|
37
|
+
import matplotlib.pyplot as plt
|
|
38
|
+
|
|
39
|
+
import numpy as np
|
|
40
|
+
|
|
41
|
+
import pandas as pd
|
|
42
|
+
|
|
43
|
+
|
|
44
|
+
|
|
45
|
+
# data = np.hstack([np.random.randint(0, 100, (30, 2)), np.random.randint(-1, 2, (30, 1))])
|
|
46
|
+
|
|
47
|
+
data = pd.read_csv("3FloorLv4_result_2021-10-20-20-47-55.csv", usecols=[0, 1, 2], nrows=1000, dtype="float")
|
|
48
|
+
|
|
49
|
+
df = pd.DataFrame(data, columns=list("XZS"))
|
|
50
|
+
|
|
51
|
+
|
|
52
|
+
|
|
53
|
+
colorList = {-1.0 : "r", 0.0 : "g", 1.0 : "b"}
|
|
54
|
+
|
|
55
|
+
labelName = {-1 : "Error", 0 : "Failure", 1 : "Success"}
|
|
56
|
+
|
|
57
|
+
|
|
58
|
+
|
|
59
|
+
for s in set(df["S"]) :
|
|
60
|
+
|
|
61
|
+
df2 = df[df["S"] == s]
|
|
62
|
+
|
|
63
|
+
c = colorList[s]
|
|
64
|
+
|
|
65
|
+
l = labelName[s]
|
|
66
|
+
|
|
67
|
+
plt.scatter(df2.X, df2.Z, color=c, label=l)
|
|
68
|
+
|
|
69
|
+
|
|
70
|
+
|
|
71
|
+
plt.legend()
|
|
72
|
+
|
|
73
|
+
plt.show()
|
|
74
|
+
|
|
75
|
+
|
|
76
|
+
|
|
77
|
+
```
|
|
78
|
+
|
|
79
|
+
|
|
80
|
+
|
|
81
|
+
すると、次のエラーが出ました。
|
|
82
|
+
|
|
83
|
+
```
|
|
84
|
+
|
|
85
|
+
Traceback (most recent call last):
|
|
86
|
+
|
|
87
|
+
File "c:\Users\yuulis\ML-Agents\EL_3Floor\DrawGragh.py", line 14, in <module>
|
|
88
|
+
|
|
89
|
+
c = colorList[s]
|
|
90
|
+
|
|
91
|
+
KeyError: nan
|
|
92
|
+
|
|
93
|
+
```
|