python – pandas：计算向下行的字符串条件

我有一个看起来像这样的数据框：

  col_1
0  A
1  A:C:D
2  A:B:C:D:E
3  B:D

我想把每个’：’计算到：

  col_1        count
0  A             0
1  A:C:D         2
2  A:B:C:D:E     4
3  B:D           1

我试过应用一个函数无济于事：

def count_fx(df):
    return df.str.contains(':').sum()
df['count'] = df['col_1'].apply(count_fx)

也,

df['count'] = df['col_1'].apply(lambda x: (x.str.contains(':')).sum(), axis=1)

解决方法:

你在这里使用apply()的方式导致每个元素被传入.即,DataFrame不是传入的.所以你只需要将你的函数改为

def count_fx(s):
    return s.count(':')

然后你可以使用你拥有的东西：

In [11]: df['count'] = df['col_1'].apply(count_fx)

In [12]: df
Out[12]:
       col_1  count
0          A      0
1      A:C:D      2
2  A:B:C:D:E      4
3        B:D      1