python – Pandas,过滤行,其中包含另一列

如何筛选包含另一列的列？
例如,如果我们有两个列A,B的DT,我们可以用B.contains(A)过滤行吗？不仅B包含来自DT的所有A的一些A值,而且只是在一行中.

A      B
'lol'  'lolec'
'ram'  'rambo'
'ki'   'pio'

Result:
A     B
'lol'  'lolec'
'ram'  'rambo'

解决方法:

您可以将boolean indexing与由apply创建的掩码一起使用,如果需要,则每行使用过滤列A和B：

#if necessary strip ' in all values
df = df.apply(lambda x: x.str.strip("'"))
#df = df.applymap(lambda x: x.strip("'"))

print (df.apply(lambda x: x.A in x.B, axis=1))
0     True
1     True
2    False
dtype: bool

df = df[df.apply(lambda x: x.A in x.B, axis=1)]
print (df)
     A      B
0  lol  lolec
1  ram  rambo

解决方案的差异 – 输入DataFrame已更改：

print (df)
     A      B
0  lol    pio
1  ram  rambo
2   ki  lolec

print (df[df.apply(lambda x: x.A in x.B, axis=1)])
     A      B
1  ram  rambo

print (df[df['B'].str.contains("|".join(df['A']))])
    A      B
1  ram  rambo
2   ki  lolec