仅当列列表包含来自其他lis的值时才筛选行

Question

我有一个数据框，其中一列是一个列表：

import pandas as pd 

data = [{'colA': 'ID1', 'colB': ['A', 'B', 'C']}, 
        {'colA': 'ID2', 'colB': ['A']}, 
        {'colA': 'ID3', 'colB': ['B']}, 
        {'colA': 'ID4', 'colB': ['A', 'B']},
        {'colA': 'ID5', 'colB': ['C', 'D']}] 

df = pd.DataFrame(data) 
df 

    colA    colB
0   ID1     [A, B, C]
1   ID2     [A]
2   ID3     [B]
3   ID4     [A, B]
4   ID5     [C, D]

我想筛选行，以便只选择那些只包含另一个列表中的值的行。但是，不应选择包含其他列表中不存在的任何值的任何行。你知道吗

valid_list = ['A', 'B']

预期结果

   colA colB
0   ID1 [A, B, C] #not selected because it contains C
1   ID2 [A] #Valid
2   ID3 [B] #Valid
3   ID4 [A, B] #Valid
4   ID5 [C, D] #Not valid - Either values not present in valid list

最终数据帧：

   colA colB
0   ID2 [A] 
1   ID3 [B] 
2   ID4 [A, B] 

Answer 1

将issubset与set一起使用：

df = df[df.colB.map(lambda x: set(x).issubset(valid_list))]

---

df = df[df.colB.map(lambda x: set(x) <= set(valid_list))]

---

print (df)
  colA    colB
1  ID2     [A]
2  ID3     [B]
3  ID4  [A, B]

Answer 2

使用np.isin和all

df[df.colB.map(lambda x: np.isin(x, valid_list).all())]

Out[217]:
  colA    colB
1  ID2  [A]
2  ID3  [B]
3  ID4  [A, B]

Answer 3

您还可以应用函数来获取遮罩：

df = df[df.colB.apply(lambda lst: all(x in valid_list for x in lst))]

---

print(df)
  colA    colB
1  ID2     [A]
2  ID3     [B]
3  ID4  [A, B]