Compare commits

..

1 commit

Author SHA1 Message Date
Danil
3486e2826a Индивидуальное задание 4
Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
2026-06-28 18:57:10 +03:00

View file

@ -1,57 +1,68 @@
import sys
import pandas as pd
import matplotlib.pyplot as plt
import sys # чтобы взять имя файла из аргументов командной строки
import pandas as pd # для чтения csv и работы с таблицей
import matplotlib.pyplot as plt # для построения гистограммы
# разбираем счёт "3:1" на два числа, пустой период считаем как 0
# Разбирает счёт вида "3:1" в одном столбце на два числовых столбца:
# голы хозяев и голы гостей. Пустой период (":") становится 0:0.
def parse_score(df, col):
parts = df[col].str.split(':', expand=True)
home = pd.to_numeric(parts[0], errors='coerce').fillna(0)
away = pd.to_numeric(parts[1], errors='coerce').fillna(0)
parts = df[col].str.split(':', expand=True) # делим каждую строку по ":" на две колонки
home = pd.to_numeric(parts[0], errors='coerce').fillna(0) # число до ":", пусто -> 0
away = pd.to_numeric(parts[1], errors='coerce').fillna(0) # число после ":", пусто -> 0
return home, away
# Индивидуальное задание №4:
# принимает имя файла и строит гистограмму общего числа заброшенных каждой командой шайб.
def task4(filename):
df = pd.read_csv(filename)
df = pd.read_csv(filename) # читаем матчи в таблицу
# счёт по периодам, овертайму и буллитам
h1, a1 = parse_score(df, 'Период_1')
h2, a2 = parse_score(df, 'Период_2')
h3, a3 = parse_score(df, 'Период_3')
oth, ota = parse_score(df, 'Овертайм')
soh, soa = parse_score(df, 'Буллиты')
# Разбираем счёт по всем стадиям матча
h1, a1 = parse_score(df, 'Период_1') # голы хозяев/гостей в 1 периоде
h2, a2 = parse_score(df, 'Период_2') # 2 период
h3, a3 = parse_score(df, 'Период_3') # 3 период
oth, ota = parse_score(df, 'Овертайм') # овертайм (0:0, если его не было)
soh, soa = parse_score(df, 'Буллиты') # буллиты (0:0, если их не было)
# Голы в основное время = сумма трёх периодов (поэлементно по строкам)
reg_h = h1 + h2 + h3
reg_a = a1 + a2 + a3
# за победу по буллитам команде добавляется одна шайба
# Ничья в основное время -> мог быть победный буллит (за него команде +1 шайба)
tie = (reg_h == reg_a)
win_home_so = tie & (oth == ota) & (soh > soa)
win_away_so = tie & (oth == ota) & (soa > soh)
win_home_so = tie & (oth == ota) & (soh > soa) # буллиты выиграли хозяева
win_away_so = tie & (oth == ota) & (soa > soh) # буллиты выиграли гости
goals_home = reg_h + oth + win_home_so.astype(int)
goals_away = reg_a + ota + win_away_so.astype(int)
# Итоговые заброшенные шайбы: основное время + овертайм + 1 за победу по буллитам
goals_home = reg_h + oth + win_home_so.astype(int) # все шайбы хозяев в матче
goals_away = reg_a + ota + win_away_so.astype(int) # все шайбы гостей в матче
# собираем шайбы обеих команд и суммируем по каждой
# Каждый матч даёт по строке двум командам: хозяевам и гостям
home = pd.DataFrame({'Команда': df['Команда_1'], 'ГЗ': goals_home})
away = pd.DataFrame({'Команда': df['Команда_2'], 'ГЗ': goals_away})
both = pd.concat([home, away], ignore_index=True)
both = pd.concat([home, away], ignore_index=True) # склеиваем оба взгляда в один список
# Суммируем заброшенные шайбы по каждой команде и сортируем по убыванию
totals = both.groupby('Команда')['ГЗ'].sum().astype(int)
totals = totals.sort_values(ascending=False)
# рисуем гистограмму
plt.figure(figsize=(12, 6))
plt.bar(totals.index, totals.values, color='steelblue')
plt.title('Всего заброшено шайб за сезон')
plt.xlabel('Команда')
plt.ylabel('Заброшено шайб')
plt.xticks(rotation=90)
for i, value in enumerate(totals.values):
teams = totals.index # ось X: названия команд
goals = totals.values # ось Y: всего заброшено шайб
plt.figure(figsize=(12, 6)) # широкое окно, чтобы влезли все команды
plt.bar(teams, goals, color='steelblue') # столбики: один на команду
plt.title('Всего заброшено шайб за сезон') # заголовок
plt.xlabel('Команда') # подпись оси X
plt.ylabel('Заброшено шайб') # подпись оси Y
plt.xticks(rotation=90) # имена команд вертикально, чтобы не накладывались
for i, value in enumerate(goals): # подпись числа над каждым столбиком
plt.text(i, value, str(value), ha='center', va='bottom')
plt.tight_layout()
plt.show()
plt.tight_layout() # подгоняем поля, чтобы подписи не обрезались
plt.show() # показываем окно с графиком
# Запуск только при прямом старте файла; имя csv берём из аргумента командной строки
if __name__ == '__main__':
task4(sys.argv[1])