27 lines
788 B
Python
27 lines
788 B
Python
|
with open('day4-input', 'r') as file:
|
||
|
|
data = sorted([l.strip('\n') for l in file])
|
||
|
|
import numpy as np
|
||
|
|
|
||
|
|
guards = {}
|
||
|
|
guard = None
|
||
|
|
guard_sleep = None
|
||
|
|
|
||
|
|
for line in data:
|
||
|
|
_, time, *event = line.split()
|
||
|
|
minute = int(time[-3:-1])
|
||
|
|
if event[0] == 'Guard':
|
||
|
|
if guard_sleep is not None:
|
||
|
|
raise Exception("Last guard didn't wake up?!")
|
||
|
|
guard = int(event[1][1:])
|
||
|
|
if guard not in guards:
|
||
|
|
guards[guard] = np.zeros((60))
|
||
|
|
elif event[0] == 'wakes':
|
||
|
|
guards[guard][guard_sleep:minute] += 1
|
||
|
|
guard_sleep = None
|
||
|
|
else: # falls asleep
|
||
|
|
guard_sleep = minute
|
||
|
|
|
||
|
|
for func in (np.sum, np.max): # Part 1, Part 2. Pattern shamelessly copied from Peter Tseng's Ruby solution.
|
||
|
|
guard, minutes = max(guards.items(), key=lambda x: func(x[1]))
|
||
|
|
print(guard * minutes.argmax())
|