【发布时间】:2016-01-29 12:52:58
【问题描述】:
我正在尝试编写一些代码,以逗号分隔数据框列中的字符串(因此它变成一个列表),并从该列表中删除某个字符串(如果存在)。删除不需要的字符串后,我想以逗号再次加入列表元素。我的数据框如下所示:
df:
Column1 Column2
0 a a,b,c
1 y b,n,m
2 d n,n,m
3 d b,b,x
所以基本上我的目标是从 column2 中删除所有 b 值,以便我得到:
df:
Column1 Column2
0 a a,c
1 y n,m
2 d n,n,m
3 d x
我写的代码如下:
df=df['Column2'].apply(lambda x: x.split(','))
def exclude_b(df):
for index, liste in df['column2].iteritems():
if 'b' in liste:
liste.remove('b')
return liste
else:
return liste
第一行将列中的所有值拆分为一个逗号分隔的列表。现在使用该函数,我尝试遍历所有列表并删除 b(如果存在),如果不存在则返回列表原样。如果我在最后打印“liste”,它只返回 Column2 的第一行,而不返回其他行。我究竟做错了什么?有没有办法将我的 if 条件实现为 lambda 函数?
【问题讨论】:
-
如何应用一个返回 Column2 而没有不需要的字符串的函数?
标签: python regex pandas lambda dataframe