def get_category()

in 07_sparkml/experiment.py [0:0]

• 9 lines of code
• 5 McCabe index (conditional complexity)

def get_category(hour):
    if hour < 6 or hour > 20:
        return [1, 0, 0]  # night
    if hour < 10:
        return [0, 1, 0]  # morning
    if hour < 17:
        return [0, 0, 1]  # mid-day
    else:
        return [0, 0, 0]  # evening