New features in the dataset

Game_of_Thrones_Feature_Engineering.py

# battles['attcker_number'] = []
@timeit
def new_features(dataframe1, dataframe2):
    ''' Creates New Features in both the dataframes
        Parameters
        -----------
        method: dataframes
        Return
        -----------
        None
    '''
    dataframe1.loc[:, 'attacker_count'] = (4 - battles[['attacker_1','attacker_2', 'attacker_3', \
                                              'attacker_4']].isnull().sum(axis=1))
    dataframe1.loc[:, 'defender_count'] = (4 - battles[['defender_1','defender_2', 'defender_3', \
                                              'defender_4']].isnull().sum(axis=1))
    dataframe1.loc[:, 'att_comm_count'] = [len(x) if type(x) == list else np.nan for \
                                        x in battles.attacker_commander.str.split(',')]
    dataframe2.loc[:, 'no_of_books'] = character_predictions[[x for x in \
                                                                character_predictions.columns \
                                                            if x.startswith('book')]].sum(axis=1)
    
new_features(battles, character_predictions)