100 lines
3.8 KiB
Python
100 lines
3.8 KiB
Python
"""plot_results.py - графики для эксперимента с лабиринтами."""
|
||
import csv
|
||
import os
|
||
import matplotlib.pyplot as plt
|
||
import numpy as np
|
||
|
||
CSV = "docs/data/results.csv"
|
||
PLOTS = "docs/data/plots"
|
||
os.makedirs(PLOTS, exist_ok=True)
|
||
|
||
|
||
def load_means():
|
||
"""Возвращает dict[(maze, strategy)] = (time_ms, visited, path_len, cost)."""
|
||
out = {}
|
||
with open(CSV, encoding="utf-8") as f:
|
||
rows = list(csv.reader(f))
|
||
start = next(i for i, r in enumerate(rows) if r and r[0] == "--- СРЕДНИЕ ---") + 2
|
||
for r in rows[start:]:
|
||
if not r:
|
||
continue
|
||
maze, strat, t, vis, plen, cost = r
|
||
out[(maze, strat)] = (float(t), int(vis), int(plen), int(cost))
|
||
return out
|
||
|
||
|
||
MAZES = ["small_10x10", "medium_51x51", "large_101x101",
|
||
"empty_30x30", "nopath_15x15",
|
||
"weighted_31x31", "weighted_choice"]
|
||
STRATEGIES = ["BFS", "DFS", "A*", "Dijkstra"]
|
||
COLORS = {"BFS": "#3498db", "DFS": "#e67e22", "A*": "#2ecc71", "Dijkstra": "#9b59b6"}
|
||
|
||
|
||
def grouped_bar(means, idx, ylabel, title, fname, log=True):
|
||
x = np.arange(len(MAZES))
|
||
w = 0.2
|
||
fig, ax = plt.subplots(figsize=(11, 5))
|
||
for i, s in enumerate(STRATEGIES):
|
||
vals = [means[(m, s)][idx] for m in MAZES]
|
||
bars = ax.bar(x + (i - 1.5) * w, vals, w, label=s, color=COLORS[s], alpha=0.9)
|
||
for b, v in zip(bars, vals):
|
||
ax.text(b.get_x() + b.get_width() / 2, b.get_height(),
|
||
f"{v:g}", ha="center", va="bottom", fontsize=7, rotation=0)
|
||
ax.set_xticks(x)
|
||
ax.set_xticklabels(MAZES, rotation=20, ha="right")
|
||
ax.set_ylabel(ylabel)
|
||
ax.set_title(title)
|
||
if log:
|
||
ax.set_yscale("log")
|
||
ax.legend()
|
||
ax.grid(axis="y", linestyle="--", alpha=0.4)
|
||
plt.tight_layout()
|
||
p = os.path.join(PLOTS, fname)
|
||
plt.savefig(p, dpi=130)
|
||
plt.close()
|
||
print("saved:", p)
|
||
|
||
|
||
def weighted_choice_chart(means):
|
||
"""Отдельный график для weighted_choice: путь vs стоимость."""
|
||
strategies = STRATEGIES
|
||
lengths = [means[("weighted_choice", s)][2] for s in strategies]
|
||
costs = [means[("weighted_choice", s)][3] for s in strategies]
|
||
x = np.arange(len(strategies))
|
||
w = 0.35
|
||
fig, ax = plt.subplots(figsize=(7.5, 4.5))
|
||
b1 = ax.bar(x - w/2, lengths, w, label="длина пути (клеток)",
|
||
color="#3498db", alpha=0.9)
|
||
b2 = ax.bar(x + w/2, costs, w, label="стоимость пути (сумма весов)",
|
||
color="#e74c3c", alpha=0.9)
|
||
for bars in (b1, b2):
|
||
for bar in bars:
|
||
ax.text(bar.get_x() + bar.get_width() / 2, bar.get_height(),
|
||
f"{bar.get_height():.0f}", ha="center", va="bottom", fontsize=9)
|
||
ax.set_xticks(x); ax.set_xticklabels(strategies)
|
||
ax.set_title("weighted_choice: BFS/DFS режут через болото,\n"
|
||
"Dijkstra/A* находят более дешёвый обход")
|
||
ax.set_ylabel("значение")
|
||
ax.legend()
|
||
ax.grid(axis="y", linestyle="--", alpha=0.4)
|
||
plt.tight_layout()
|
||
p = os.path.join(PLOTS, "weighted_choice_compare.png")
|
||
plt.savefig(p, dpi=130)
|
||
plt.close()
|
||
print("saved:", p)
|
||
|
||
|
||
def main():
|
||
means = load_means()
|
||
grouped_bar(means, 0, "Время, мс (среднее по 7 запускам, лог. шкала)",
|
||
"Время поиска пути", "time_compare.png", log=True)
|
||
grouped_bar(means, 1, "Число посещённых клеток (лог. шкала)",
|
||
"Сколько клеток посетил алгоритм", "visited_compare.png", log=True)
|
||
grouped_bar(means, 2, "Длина пути (клеток)",
|
||
"Длина найденного пути", "path_compare.png", log=False)
|
||
weighted_choice_chart(means)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|