pandas 两个DataFrame带条件匹配

题目

给定两个DataFrame,第一个D1是大学的全称,第二个D2是大学的缩写,如果D2里面有D1大学的缩写,就在对应的D1后面加一个“yes”。
例:
D1
pandas 两个DataFrame带条件匹配_第1张图片
D2
pandas 两个DataFrame带条件匹配_第2张图片

最后实现结果
pandas 两个DataFrame带条件匹配_第3张图片

要求

用一行代码实现上述功能

实现方法

  1. 暴力解法
import pandas as pd
import re

# for example
d1 = pd.DataFrame(['上海大学', '上海理工大学', '华东师范大学'], columns=['school'])
d2 = pd.DataFrame(['上理工','上交大','华师大'],columns=['abb'])

# add a column, set values 'no'
d1['name'] = ['no' for i in range(d1.shape[0])]

# loop d1
for i in range(d1.shape[0]): 
    # loop d2
    for k in range(d2.shape[0]):
        # transform d2 to re  
        s = '' 
        r = d2.iloc[k,0]
        for j in range(len(r)):
            s += r[j]
            s += '.{0,}'
            
        if len(re.findall(s,d1.iloc[i,0])) !=0:
            d1.loc[i,'name'] = 'yes'
            break
  1. 一行代码实现
d1['name'] = np.where(d1['school'].apply(lambda y: any(re.search(x,y) for x in d2['abb'].apply(lambda x:'.*'.join(list(x))))),'YES','NO')

你可能感兴趣的:(数据结构与算法)