python - Pandas - 按 id 分组并删除重复项

Question

我有以下数据。

userid itemid timestamp
  1       1      50
  1       2      50
  1       3      50
  1       4      60
  2       1      40
  2       2      50

我想删除具有重复时间戳的用户。在上面的示例中，我想删除 userid=1，因为他在时间戳=50 时查看了多个项目。 userid=2 不应被删除，因为他查看了不同时间戳的项目。

谁能帮我吗？

Answer 1

选项 1
使用重复

dropid = df.loc[
    df.duplicated(subset=['userid', 'timestamp']),
    'userid'
].unique()
df[~df.userid.isin(dropid)]

选项 2
使用groupby和filter

df.set_index(['userid', 'timestamp']).groupby(level=0).filter(
    lambda x: ~x.index.is_unique
).reset_index()