example_catagory

ex.py

def catagorize(row):
    """
    apply function to catagorize a set of traing data values into 
    catagories. 
    """
    sentiment = 0
    focus = 0 
    energy_level = 0 ## TODO replace with real
    if sentiment > 0:
        if focus > 0: # focus outward
            if energy_level > .25:
                return "connected"
            elif energy_level < -.25:
                return "tolerant"
            else:
                return "vunerable"
        else: # focus inward
            if energy_level > .25:
                return "hopeful" 
            elif energy_level < -.25:
                return "calm" 
            else:
                return "happy"
    elif sentiment <= 0:
        if focus > 0: #focus outward
            if energy_level > .25:
                return "angry"
            elif energy_level < -.25:
                return "lonely" 
            else:
                return "guarded"
        else: #focus inward 
            if energy_level > .25:
                return  "anxious" 
            elif energy_level < -.25:
                return "sad"
            else:
                return "fearful"