主页 / 问题 / 1213888
Accepted
Владислав Харламов
Asked:2021-12-04 16:25:44 +0000 UTC2021-12-04 16:25:44 +0000 UTC

按唯一组合分组和过滤

  • 772

有一个这样的表：

user, cart
1, 1
1, 1
1, 2
2, 1
2, 2
2, 3
2, 3
2, 3
3, 1
...

你怎么能只得到那些 yuser至少有两个 unique的行cart，并且只有那些cart具有最小数字的行。也就是说，对于给定的表，我想得到：

user, cart
1, 1
1, 1
2, 1

user=1走近，因为 他有cart=1,2，所以我们离开他，但只取cart最小的数字，即 1，所以只有：

1, 1
1, 1

同样对于user=2, 因为user=3我们不取 , 因为 只有一个唯一的cart等于 1。

python

1 个回答

  1. Best Answer
    res = (df
       .merge(df
              .groupby("user")
              .filter(lambda x: x["cart"].nunique() > 1)
              .groupby("user")
              ["cart"]
              .min()
              .reset_index()))

    结果：

    In [407]: res
Out[407]: 
   user  cart
0     1     1
1     1     1
2     2     1
    • 5

相关问题