我有一个熊猫数据框,显示用户在测试问题上的表现.看起来像这样:
userID questionID correct
-------------------------------
1 1 1
1 5 1
1 6 0
1 8 0
1 10 1
2 3 1
2 5 1
2 6 0
. . .
. . .
. . .
我想为每个用户创建一个特征向量,说明他们是否正确回答了每个问题,看起来像这样:
questionID 1 2 3 4 5 6 ...
userID -------------------------------------------------
1 1 NaN NaN NaN 1 0 ...
2 NaN NaN 1 NaN 1 0 ...
. ...
. ...
.
仅向每个用户显示所有问题的子集,因此它是一个稀疏矩阵.
如何用熊猫制作上表?
我想做下面的事情-按用户ID和问题ID分组,然后拆栈,但是我不确定它应该如何工作.
df = df.groupby(['user_id','question_id'])
df.unstack()
谢谢你的帮助.