检查每一行的另一列中是否存在数据框中的值

时间:2020-09-30 21:52:39

标签: python pandas numpy dataframe

有问题的数据框df:

colA colB
1  [1, 4, 5]
4  [3, nan, nan]

我试图返回具有True的Series,其中colA的值在每一行的colB的值中。

结果应为:

True
False

我尝试过:df.colA.isin(df.colB)-但这不能解决问题,因为colB的值在列表中

1 个答案:

答案 0 :(得分:1)

您需要解压缩isin之前的列表列

m = pd.DataFrame(df['colB'].tolist(),index=df.index).isin(df['colA']).any(axis=1)