Pandas 数据帧检查列是否包含存在于另一列中的字符串



我正在尝试学习数据帧,但我仍然是初学者。 假设我有一个包含两列的数据帧:

Name       Description
Am         Owner of Am
BQ         Employee at BQ  
JW         Employee somewhere

我想检查名称是否也是描述的一部分,如果是,请保留该行。如果不是,请删除该行。在这种情况下,它将删除第 3 行(某处的 JW 员工(

试试这个:

df[df.apply(lambda x: x['Name'] in x['Description'], axis = 1)]
s='|'.join(df.Name)#Join the search words into a pattern
df=df[df.Description.str.contains(s)]#Mask using boolean select
print (df)
Name     Description
0   Am     Owner of Am
1   BQ  Employee at BQ

%%timeit
s='|'.join(df.Name)
df[df.Description.str.contains(s)]
537 µs ± 2.37 µs per loop (mean ± std. dev. of 7 runs, 1000 loops each)
%timeit df[df.apply(lambda x: x['Name'] in x['Description'], axis = 1)]
1.27 ms ± 3.22 µs per loop (mean ± std. dev. of 7 runs, 1000 loops each)

最新更新