69 lines
1.3 KiB
Python
69 lines
1.3 KiB
Python
import json
|
|
|
|
import matplotlib.pyplot as plt
|
|
|
|
with open("./res.txt") as res_file:
|
|
res = json.load(res_file)
|
|
|
|
good_len = -1
|
|
bad_len = -1
|
|
with open("../fortune_generator/json/fortune.json") as fortune_file:
|
|
j = json.load(fortune_file)
|
|
good_len = len(j["goodFortunes"])
|
|
bad_len = len(j["badFortunes"])
|
|
|
|
status_bucket = [0] * 8
|
|
good_bucket = [0] * good_len
|
|
bad_bucket = [0] * bad_len
|
|
|
|
for ip, v in res.items():
|
|
assert all(val != -1 for val in v)
|
|
good_bucket[v[0]] += 1
|
|
good_bucket[v[1]] += 1
|
|
bad_bucket[v[2]] += 1
|
|
bad_bucket[v[3]] += 1
|
|
status_bucket[v[4]] += 1
|
|
|
|
|
|
groups = 1
|
|
fig, axs = plt.subplots(groups, 1, figsize=(8, 6))
|
|
|
|
axs.bar(
|
|
range(good_len),
|
|
good_bucket,
|
|
color="skyblue",
|
|
edgecolor="black",
|
|
)
|
|
axs.set_xlabel("Good Fortune Event Index")
|
|
axs.set_ylabel("Occurrences")
|
|
|
|
plt.tight_layout()
|
|
plt.show()
|
|
|
|
fig, axs = plt.subplots(groups, 1, figsize=(8, 6))
|
|
|
|
axs.bar(
|
|
range(bad_len),
|
|
bad_bucket,
|
|
color="skyblue",
|
|
edgecolor="black",
|
|
)
|
|
axs.set_xlabel("Bad Fortune Event Index")
|
|
axs.set_ylabel("Occurrences")
|
|
|
|
plt.tight_layout()
|
|
plt.show()
|
|
|
|
fig, axs = plt.subplots(groups, 1, figsize=(8, 6))
|
|
|
|
axs.bar(
|
|
range(len(status_bucket)),
|
|
status_bucket,
|
|
color="skyblue",
|
|
edgecolor="black",
|
|
)
|
|
axs.set_xlabel("Status Index")
|
|
axs.set_ylabel("Occurrences")
|
|
|
|
plt.show()
|