from reader import Reader
from visualizer import Visualizer
from kmeans import KMeans
if __name__ == '__main__':
reader = Reader('../data/crime.csv')
visualizer = Visualizer(reader)
# Question No.1
visualizer.countplot('YEAR', 'Crimes per Year')
visualizer.countplot('MONTH', 'Crimes per Month')
visualizer.countplot('DAY_OF_WEEK', 'Crimes per Day')
visualizer.countplot('DISTRICT', 'Crimes per District')
# Question No.2
visualizer.countplot('YEAR', 'Shootings per Year', predicate=lambda data: data['SHOOTING'] == 'Y')
visualizer.countplot('DISTRICT', 'Shootings per District', predicate=lambda data: data['SHOOTING'] == 'Y')
# Question No.3
visualizer.countplot('TIME_PERIOD', 'Crimes per Time Period')
# Question No.4
visualizer.countplot('OFFENSE_CODE_GROUP', 'Types Of Crime During The Day', predicate=lambda data: data['TIME_PERIOD'] == 'Day', squeeze=True)
