forked from zovos/summer_practice
79 lines
3.3 KiB
Python
79 lines
3.3 KiB
Python
import os
|
||
import sys
|
||
|
||
import matplotlib.pyplot as plt
|
||
import pandas as pd
|
||
|
||
import main as mn
|
||
|
||
def calculate_team_points(filename):
|
||
df = pd.read_csv(filename)
|
||
# Считаем очки команд за периоды, овертаймы и булиты
|
||
h1, a1 = mn.parse_score(df, 'Период_1')
|
||
h2, a2 = mn.parse_score(df, 'Период_2')
|
||
h3, a3 = mn.parse_score(df, 'Период_3')
|
||
oth, ota = mn.parse_score(df, 'Овертайм')
|
||
soh, soa = mn.parse_score(df, 'Буллиты') # Счет по буллитам
|
||
|
||
# Складываем очки, получая итоговый счет
|
||
home_goals = h1 + h2 + h3
|
||
away_goals = a1 + a2 + a3
|
||
tie = home_goals == away_goals
|
||
|
||
# Проверка кто проиграл/выйграл по итогу матча или овертайма, или буллита
|
||
home_win = (home_goals > away_goals) | (tie & ((oth > ota) | ((oth == ota) & (soh > soa))))
|
||
away_win = (away_goals > home_goals) | (tie & ((ota > oth) | ((oth == ota) & (soa > soh))))
|
||
|
||
home_overtime_loss = tie & ((ota > oth) | ((oth == ota) & (soa > soh)))
|
||
away_overtime_loss = tie & ((oth > ota) | ((oth == ota) & (soh > soa)))
|
||
|
||
# Подсчитываем итоговое кол-во очков
|
||
home_points = 2 * home_win.astype(int) + home_overtime_loss.astype(int)
|
||
away_points = 2 * away_win.astype(int) + away_overtime_loss.astype(int)
|
||
|
||
# Соединяем 2 таблицы
|
||
points = pd.concat([
|
||
pd.DataFrame({'Команда': df['Команда_1'], 'О': home_points}),
|
||
pd.DataFrame({'Команда': df['Команда_2'], 'О': away_points}),
|
||
])
|
||
|
||
# Собираем таблицу с помощью groupby,
|
||
return points.groupby('Команда')['О'].sum().sort_values(ascending=False)
|
||
|
||
|
||
def show_points_histogram(filename, output=None):
|
||
points = calculate_team_points(filename) # Получаем очки
|
||
|
||
plt.figure(figsize=(12, 7)) # Создаем таблицу 12 на 7 дюймов
|
||
plt.bar(points.index, points.values, color='steelblue') # Строим столбчатую диаграмму, по очкам по y
|
||
plt.title('Гистограмма набранных командами очков')
|
||
plt.xlabel('Команда')
|
||
plt.ylabel('Очки')
|
||
plt.xticks(rotation=75, ha='right') # Чтобы не накладывались подписи, мы выравниваем и вращаем их
|
||
plt.grid(axis='y', alpha=0.3) # Отображение координатной сетки
|
||
plt.tight_layout() # Автоматическая настройка отступов
|
||
|
||
# Сохранение если указан нужный аргумент
|
||
if output:
|
||
plt.savefig(output, dpi=200)
|
||
print(f'Гистограмма сохранена в файл: {output}')
|
||
else:
|
||
plt.show()
|
||
|
||
|
||
def main():
|
||
if len(sys.argv) not in (2, 3):
|
||
print('Использование: python3 individual_1.py <csv-файл> [файл_для_сохранения]')
|
||
sys.exit(1)
|
||
|
||
filename = sys.argv[1]
|
||
output = sys.argv[2] if len(sys.argv) == 3 else None
|
||
|
||
if not os.path.isfile(filename):
|
||
raise FileNotFoundError(f'Файл не найден: {filename}')
|
||
|
||
show_points_histogram(filename, output)
|
||
|
||
|
||
if __name__ == '__main__':
|
||
main()
|