Python 提取一个新的数据框

Question

我有一个数据框:

  topic  student level week
    1      a       1     1
    1      b       2     1
    1      a       3     1
    2      a       1     2
    2      b       2     2
    2      a       3     2
    2      b       4     2
    3      c       1     2
    3      b       2     2
    3      c       3     2
    3      a       4     2
    3      b       5     2

我想提取主题中学生的一些消息并创建一个包含三列的新 df:

  student  topic  messages
     a       1        2
     a       2        2
     a       3        1
     b       1        1
     b       2        2
     b       3        2
     c       3        2

我想跳过包含 0 条消息的行。

有人有什么建议吗？

谢谢!

Answer 1

您可以使用 groupby + size + reset_index :

df = df.groupby(['student','topic']).size().reset_index(name='messages')
print (df)
  student  topic  messages
0       a      1         2
1       a      2         2
2       a      3         1
3       b      1         1
4       b      2         2
5       b      3         2
6       c      3         2