Lester / Uni WPS Data

output.csv

parser.py

import csv

files = ['WRPSOccurrenceData_Year_2011.csv',
'WRPSOccurrenceData_Year_2012.csv',
'WRPSOccurrenceData_Year_2013.csv',
'WRPSOccurrenceData_Year_2014.csv',
'WRPSOccurrenceData_Year_2015.csv',
'WRPSOccurrenceData_Year_2016_Jan16_2018.csv',
'WRPSOccurrenceData_Year_2017__August_28_2018.csv']
year = 2011
histo = dict()

with open('output.csv', 'w', newline='') as csvfile:
    csvwriter = csv.writer(csvfile, delimiter=',')
    csvwriter.writerow(['Year', 'Intersection', 'Incident', 'Date'])
    for f in files :
        with open(f) as csv_file:
            print(f'YEAR: {year}')
            histo[year] = 0
            csv_reader = csv.reader(csv_file, delimiter=',')
            line_count = 0
            for row in csv_reader:
                if line_count == 0:
                    #print(f'Column names are {", ".join(row)}')
                    line_count += 1
                else:
                    inters = row[2].upper()
                    report_type_1 = row[10]
                    report_type_2 = row[8]

                    if (("LESTER ST / UNIVERSITY AVE W" in inters or \
                    "SUNVIEW ST / UNIVERSITY AVE W" in inters or \
                    "ALBERT ST / UNIVERSITY AVE W" in inters) and ( \
                    "9530" in report_type_1 or \
                    "9510" in report_type_1 or \
                    "9520" in report_type_1 or \
                    "9530" in report_type_2 or \
                    "9510" in report_type_2 or \
                    "9520" in report_type_2)):
                        print(f'\t{row[2]} - {row[6]} - {row[10]}.')
                        csvwriter.writerow([year, row[2], row[10], row[6]])
                        histo[year] += 1
                    line_count += 1
            print(f'Processed {line_count} lines.')
            year += 1

print(histo)