如何在pandas DataFrame的列表类型列中删除指定字符串
原代码错误原因
- 语法不符合Python规范:Python中没有
if ... then的写法,lambda内的三元判断语法为返回值1 if 条件 else 返回值2 - 列表方法特性错误:
list.remove()是原地修改操作,执行后返回None,运行后你得到的列会全部是空值,无法拿到修改后的列表 - 赋值逻辑错误:直接将单列map的结果赋值给
df,会把原本的多列DataFrame替换为单列Series,丢失Column1等其他列数据
正确实现方案
推荐使用列表推导式直接生成过滤后的新列表,逻辑更简洁,也不会触发原地修改的问题,该写法会删除列表中所有的'b':
df['Column2'] = df['Column2'].map(lambda x: [item for item in x if item != 'b'])
如果你只需要删除列表中第一个出现的'b'(匹配原生remove()方法的删除逻辑),可以用自定义函数实现:
def drop_first_b(lst): new_lst = lst.copy() if 'b' in new_lst: new_lst.remove('b') return new_lst df['Column2'] = df['Column2'].map(drop_first_b)
结果验证
执行后你的DataFrame输出如下:
| Column1 | Column2 |
|---|---|
| 1 | ['a'] |
| 2 | [] |
内容的提问来源于stack exchange,提问作者tester559
相关产品推荐
相关产品推荐

