python - 根据另一个列表从熊猫数据框列中的列表中删除值

Question

我在包含列表的数据框中有一列。我希望能够根据我在另一个列表中的元素从这些列表中删除元素（如下所示）。

我尝试使用列表理解，但似乎没有结果。

import pandas as pd

sys_list = ['sys1', 'sys2', 'sys3']
df = pd.DataFrame({'A':[['sys1', 'sys2', 'user1'], 
                        ['user3', 'user6', 'user1'], 
                        ['sys1', 'sys2', 'sys3']]})

df['A'] = [item for item in df['A'] if item not in sys_list]

print(df)

                       A
0    [sys1, sys2, user1]
1  [user3, user6, user1]
2     [sys1, sys2, sys3]

我需要实现这一点：

                       A
0                [user1]
1  [user3, user6, user1]
2                     []

有什么想法吗？

标签： pythonpandasdataframelist-comprehension

Answer 1

使用Series.apply：

df['B'] = df['A'].apply(lambda x: [item for item in x if item not in set(sys_list)])
print (df)
                       A                      B
0    [sys1, sys2, user1]                [user1]
1  [user3, user6, user1]  [user3, user6, user1]
2     [sys1, sys2, sys3]                     []

或类似的列表理解，如已删除的答案：

df['B'] = [[item for item in l if item not in set(sys_list)] for l in df['A']]

或使用sets 的解决方案set.difference：

df['B'] = df['A'].map(set(sys_list).difference).map(list)