63 lines
2.1 KiB
Python
63 lines
2.1 KiB
Python
import csv
|
||
import statistics
|
||
import matplotlib.pyplot as plt
|
||
import numpy as np
|
||
|
||
|
||
rows = []
|
||
with open("results.csv", encoding="utf-8") as f:
|
||
for r in csv.DictReader(f):
|
||
rows.append(r)
|
||
|
||
MAZES = ["small_10x10", "medium_50x50", "large_100x100",
|
||
"empty_30x30", "no_exit_20x20", "weighted_40x40"]
|
||
STRATS = ["BFS", "DFS", "A*", "Dijkstra"]
|
||
MAZE_RU = {
|
||
"small_10x10": "10×10",
|
||
"medium_50x50": "50×50",
|
||
"large_100x100": "100×100",
|
||
"empty_30x30": "30×30 пустой",
|
||
"no_exit_20x20": "20×20 без выхода",
|
||
"weighted_40x40":"40×40 взвешенный",
|
||
}
|
||
|
||
def avg(maze, strat, metric):
|
||
vals = [float(r[metric]) for r in rows
|
||
if r["maze"] == maze and r["strategy"] == strat]
|
||
return statistics.mean(vals) if vals else 0.0
|
||
|
||
def std(maze, strat, metric):
|
||
vals = [float(r[metric]) for r in rows
|
||
if r["maze"] == maze and r["strategy"] == strat]
|
||
return statistics.stdev(vals) if len(vals) > 1 else 0.0
|
||
|
||
|
||
fig, axes = plt.subplots(1, 3, figsize=(16, 5))
|
||
fig.suptitle("Сравнение алгоритмов (среднее, 7 запусков)")
|
||
|
||
x = np.arange(len(MAZES))
|
||
W = 0.18
|
||
offsets = np.linspace(-(len(STRATS)-1)/2, (len(STRATS)-1)/2, len(STRATS)) * W
|
||
|
||
for ax, (metric, ylabel, title) in zip(axes, [
|
||
("time_ms", "Время (мс)", "Время выполнения"),
|
||
("visited_cells", "Посещено клеток", "Посещённые клетки"),
|
||
("path_length", "Длина пути", "Длина найденного пути"),
|
||
]):
|
||
for i, strat in enumerate(STRATS):
|
||
vals = [avg(m, strat, metric) for m in MAZES]
|
||
errs = [std(m, strat, metric) for m in MAZES]
|
||
ax.bar(x + offsets[i], vals, W * 0.95, label=strat, yerr=errs, capsize=3)
|
||
ax.set_title(title)
|
||
ax.set_xticks(x)
|
||
ax.set_xticklabels([MAZE_RU[m] for m in MAZES], fontsize=7, rotation=15)
|
||
ax.set_ylabel(ylabel)
|
||
ax.legend(fontsize=8)
|
||
ax.yaxis.grid(True, linestyle="--", alpha=0.5)
|
||
ax.set_axisbelow(True)
|
||
if metric == "time_ms":
|
||
ax.set_yscale("log")
|
||
|
||
plt.tight_layout()
|
||
plt.savefig("../../performance_plot.png", dpi=150)
|
||
plt.close() |