df_A,df_B都有一列‘col',为了把df_A中col列中元素也在df_B的找到,先写了

lst = list(df_B['col'])

df_A.loc[ df_A.apply(lambda line: line['col'] in lst) ]

但这运行速度非常慢。

pandas中有个函数isin,能快速实现这个目的。

df_A.loc[ df_A['col'].isin(df_B['col']) ]

查看原文 >>
相关文章