我正在尝试学习数据帧,但我仍然是初学者。 假设我有一个包含两列的数据帧:
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)