s9_merge_final_datafiles.py 文件源码

def get_same_status(pairs, items, target):
    text_compare = pairs
    item1 = items[['itemID', target]]
    item1 = item1.rename(
        columns={
            'itemID': 'itemID_1',
            target: target + '_1',
        }
    )
    text_compare = pd.merge(text_compare, item1, how='left', on='itemID_1', left_index=True)
    item2 = items[['itemID', target]]
    item2 = item2.rename(
        columns={
            'itemID': 'itemID_2',
            target: target + '_2',
        }
    )
    text_compare = pd.merge(text_compare, item2, how='left', on='itemID_2', left_index=True)
    text_compare[target + '_same'] = np.equal(text_compare[target + '_1'], text_compare[target + '_2']).astype(np.int32)
    # print(text_compare[target + '_same'].describe())
    return text_compare[['id', target + '_same']]