如何比较两列并从第三列返回值 Pandas dataframe

Question

样本数据

sample = ({'name':['Delinquency Rate','Cumulative Probability'],
           'value':['Dlnqy','Prbblty'],
           'new_name':['Dlnqncy Rt','Cmltv Prbblty']})

test = pd.DataFrame(sample)

test

new_name 是通过从“name”中删除所有元音创建的。 我想比较“新名称”中“值”的前 3 个字符和“每个单词”的前 3 个字符，如果它们匹配，我想从名称列中返回值。 例如'Dln' in value 存在于new_name 的'Dlnqncy' 中，所以我们将从name 中返回'Delinquency'。 逻辑应该像

'if value[:3] in new_name[:3] then return name'

以下是我到目前为止所拥有的，如果我只有两列，它就可以正常工作。 但是如果我想比较 value 和 new_name 并重新调整名称，它就不起作用了。

def get_matches(name, value, new_name, default=''):
    return next( (word for word in new_name.split() if str(value)[:3] in word[:3]),default)


test['match'] = test[['name', 'value', 'new_name']].apply(lambda row: get_matches(*row, default=' '), axis=1)

在下表中，“匹配”列应该有“拖欠”和“概率”（因为它在“名称”列中）

Answer 1

给你go：

(test
 .assign(match=lambda x: np.where((x.new_name.str[:3] == x.value.str[:3]), x.name, x.value))
)