首页 > 解决方案 > 在Python中按多列分组时如何为组设置条件?

问题描述

我有一个按 atc、product_name、strength 和 form 分组的 DataFrame,如随附的屏幕截图所示。我希望代码返回有超过 1 个唯一替换组的组。

分组数据框

我已经尝试了下面的代码,但它也返回只有 1 个唯一组的组:

df.loc[df.groupby(['atc','product_name','strength','form'])['substitution_group'].apply(lambda x: len(x.unique()) >= 2)]

此外,Groupby 不会将相同的名称和优势归为一组,而是将多个组放在一起。有针对这个的解决方法吗?

标签: pythonpandas-groupby

解决方案


推荐阅读