2026-rff_mp/GutovVM/docs/data/lab_2_data/graphics.py

29 lines
941 B
Python
Raw Normal View History

2026-05-26 22:45:20 +00:00
#Графики
import matplotlib.pyplot as plt
2026-05-28 14:54:18 +00:00
import pandas as pd
2026-05-26 22:45:20 +00:00
2026-05-28 14:54:18 +00:00
df = pd.read_csv('results.csv', header=None, names=['lab','strategy','timeMs','cellsVisited','pathLength'])
2026-05-26 22:45:20 +00:00
2026-05-28 14:54:18 +00:00
print(df)
2026-05-26 22:45:20 +00:00
2026-05-28 14:54:18 +00:00
df_time = df.pivot(index='lab', columns='strategy', values=['timeMs','cellsVisited','pathLength'])
2026-05-26 22:45:20 +00:00
2026-05-28 14:54:18 +00:00
print(df_time)
2026-05-26 22:45:20 +00:00
2026-05-28 14:54:18 +00:00
# 1. График только для Времени
df_time["timeMs"].plot(kind="bar", figsize=(10, 5), rot=0)
plt.title("Время работы стратегий (мс)")
plt.ylabel("timeMs")
plt.show()
2026-05-26 22:45:20 +00:00
2026-05-28 14:54:18 +00:00
# 2. График для Посещенных клеток
df_time["cellsVisited"].plot(kind="bar", figsize=(10, 5), rot=0)
plt.title("Количество посещенных клеток")
plt.ylabel("cellsVisited")
plt.show()
# 3. График для Длины пути
df_time["pathLength"].plot(kind="bar", figsize=(10, 5), rot=0)
plt.title("Длина найденного пути")
plt.ylabel("pathLength")
plt.show()