pandas groupby count字符串出现在列上
pandas groupby count字符串出现在列上
提问于 2015-07-27 17:37:17
我想计算一个分组的pandas dataframe列中字符串的出现次数。
假设我有以下数据帧：
catA catB scores
A X 6-4 RET
A X 6-4 6-4
A Y 6-3 RET
B Z 6-0 RET
B Z 6-1 RET首先，我想按catA和catB分组。对于这些组中的每个组，我希望在scores列中计算RET的出现次数。
结果应该如下所示：
catA catB RET
A X 1
A Y 1
B Z 2按两列分组很容易：grouped = df.groupby(['catA', 'catB'])
但是下一步呢？
回答 1
Stack Overflow用户
回答已采纳
发布于 2015-07-27 17:43:03
对groupby对象的'scores‘列调用apply，并使用矢量化str方法contains，使用此方法过滤group并调用count
In [34]:
df.groupby(['catA', 'catB'])['scores'].apply(lambda x: x[x.str.contains('RET')].count())
Out[34]:
catA catB
A X 1
Y 1
B Z 2
Name: scores, dtype: int64要将其赋值为列，请使用transform，以便聚合返回索引与原始df对齐的序列：
In [35]:
df['count'] = df.groupby(['catA', 'catB'])['scores'].transform(lambda x: x[x.str.contains('RET')].count())
df
Out[35]:
catA catB scores count
0 A X 6-4 RET 1
1 A X 6-4 6-4 1
2 A Y 6-3 RET 1
3 B Z 6-0 RET 2
4 B Z 6-1 RET 2页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/31649669
复制相关文章
相似问题
腾讯云开发者
