如何根据外部列表中的值过滤熊猫数据框?

时间:2020-08-09 14:47:23

标签: python python-3.x pandas

我有以下数据框。我想将layout_collapseParallaxMultiplier列中每个逗号分隔的值与a中的值进行比较,如果values列中存在任何值,则返回True 。希望您能理解这种情况。谢谢!

a

我希望这将返回:

import pandas as pd
values = ['value 1', 'value 2']

df = pd.DataFrame({'a': ['value 1', 'value 1, value 2', 'value 1, value 3', 'value 3', 'value 4', 'value 5']})

                  a
0           value 1
1  value 1, value 2
2  value 1, value 3
3           value 3
4           value 4
5           value 5

1 个答案:

答案 0 :(得分:0)

让我们尝试splitany

df['b']=df.a.str.split(', ',expand=True).isin(values ).any(1)
df
Out[168]: 
                  a      b
0           value 1   True
1  value 1, value 2   True
2  value 1, value 3   True
3           value 3  False
4           value 4  False
5           value 5  False