mlb_data_analysis/BattingDataDisplay.py at main · blainefreestone/mlb_data_analysis · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
from BattingData import BattingData

import matplotlib.pyplot as plt

class BattingDataDisplay:
    def __init__(self):
        self.batting_data = BattingData()

    def graph_player_batting_statistics(self, player_id):
        player_data = self.batting_data.statistics_for_player(player_id)
        player_data.plot(marker="o", linestyle="dashed")
        plt.title(f"{self.batting_data.name_for_player(player_id)} Statistics")
        plt.xlabel("Season")
        plt.show()

    def graph_league_batting_statistics(self):
        league_data = self.batting_data.statistics_for_league()
        league_data.plot(marker="none")
        plt.title("MLB League Statistics 1871-2015")
        plt.xlabel("Season")
        plt.show()

    def print_player_batting_statistics(self, player_id):
        print(self.batting_data.statistics_for_player(player_id))

    def print_league_batting_statistics(self):
        print(self.batting_data.statistics_for_league())

    def print_player_batting_data(self, player_id):
        print(self.batting_data.for_player(player_id))

    def print_league_batting_data(self):
        print(self.batting_data.for_league())

    def print_predict_model_data(self):
        print(self.batting_data.for_predict_model())