如何移除Pandas DataFrame列表列中不含下划线的元素？

原始数据 #

import pandas as pd
w = pd.DataFrame({'Column A': ['Cell 1', 'Cell 2', 'Cell 3'], 'Column B': [['a_1', 'a'], ['b1', 'b2'], ['c_1','c_2', 'c3']]})

期望结果 #

Column A        Column B
0   Cell 1        [a_1]
1   Cell 2        []
2   Cell 3        [c_1, c_2]

实现方法 #

• 方法一：lambda函数快速筛选
  直接用apply遍历Column B的每个列表，通过列表推导式只保留包含下划线的元素：

w['Column B'] = w['Column B'].apply(lambda x: [item for item in x if '_' in item])

• 方法二：自定义函数（可读性更强）
  如果需要更清晰的代码结构，可单独定义筛选函数再通过apply调用：

def keep_underscore_items(lst):
    return [item for item in lst if '_' in item]

w['Column B'] = w['Column B'].apply(keep_underscore_items)

运行上述任意一种方法后，打印w即可得到目标结果。

内容的提问来源于stack exchange，提问作者lb0389