Source code for src.FamilyTable

#!/usr/bin/env python
import pandas as pd
import numpy as np

class Family(pd.DataFrame):
    def __init__(self, *args, **kwargs):
        super(Family, self).__init__(*args, **kwargs)
        
    def update_income(self):
        """
        function updates income according to changes in wage or hrs/week
        """
        wrk = self.Hieu.Worker
        hieu = self.Hieu
        #map workers to families
        for fam in np.unique(self.family_id):
            #get all members of family in worker table
            workers = wrk[wrk.person_id.isin(self.Hieu[self.Hieu.family_id == fam].person_id)]
            self.loc[self.family_id == fam, "family_income"] = sum(workers.hourly_wage * workers.hrs_week * 52)