我有一个数据框:
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 条消息的行。
有人有什么建议吗？
谢谢!
最佳答案
您可以使用 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
关于Python 提取一个新的数据框，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/43820340/