计算熊猫群中的唯一值

Question

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

data = {'id': [1,1,1,2,2,3],
        'value': ['a','a','a','b','b','c'],
        'obj_id': [1,2,3,3,3,4]
}
df = pd.DataFrame (data, columns = ['id','value','obj_id'])

我想通过obj_id和id获得value组的唯一计数：

1 a 3
2 b 1
3 c 1

但是当我这样做时：

result=df.groupby(['id','value'])['obj_id'].nunique().reset_index(name='obj_counts')

我得到的结果是：

1 a 2
1 a 1
2 b 1
3 c 1

因此具有相同id和value的前两行不会分组在一起。

我该如何解决？非常感谢！

Answer 1

对我来说，您的解决方案可以很好地处理示例数据。

就像在评论中提到@YOBEN_S一样，可能是解决白人问题，然后解决方案是添加Series.str.strip：

Series.str.strip