从熊猫数据框中的列表中删除元素

Question

我想从 pandas 数据框列的列表中删除特定值。我该怎么办？

df = pd.DataFrame({'A': ['a1', 'a2', 'a3'],
               'B': [['b1', 'b2'], ['b1', 'b1'], ['b2']],
               'C': [['c1', 'b1'], ['b3'], ['b2', 'b2']],
               'D': ['d1', 'd2', 'd3']})

我想从数据框中的“B”列中删除列表中的字符串“b1”。

Answer 1

你可以使用列表推导

df['B'] = df.B.apply(lambda x: [i for i in x if i != 'b1'])


    A   B       C           D
0   a1  [b2]    [c1, b1]    d1
1   a2  []      [b3]        d2
2   a3  [b2]    [b2, b2]    d3

Answer 2

和我之前的所有回答一样

(df.B.map(set)-{'b1'}).map(list)
Out[171]: 
0    [b2]
1      []
2    [b2]
Name: B, dtype: object