python - 基于另一列从 pandas 数据框中的列中的每个列表中删除元素

Question

我想根据 A 列从 B 列中删除列表中的值，不知道如何操作。

给定:

df = pd.DataFrame({
    'A': ['a1', 'a2', 'a3', 'a4'],
    'B': [['a1', 'a2'], ['a1', 'a2', 'a3'], ['a1', 'a3'], []]
})

我想要:

result = pd.DataFrame({
    'A': ['a1', 'a2', 'a3', 'a4'],
    'B': [['a1', 'a2'], ['a1', 'a2', 'a3'], ['a1', 'a3'], []],
    'Output': [['a2'], ['a1', 'a3'], ['a1'], []]
})

Answer 1

一种方法是通过 DataFrame.apply 对每一行应用过滤函数:

df['Output'] = df.apply(lambda x: [i for i in x.B if i != x.A], axis=1)