python - 将交叉表值与数据框连接起来？

Question

我有以下类型的数据框 df:

ID   Result Other_val
1    A      y
2    B      x
2    A      x
3    C      abc

使用 pd.crosstab(df.ID, df.Result) 后，我得到了这样的交叉表:

Result  A  B  C
ID
1       1  0  0
2       1  1  0
3       0  0  1

我现在希望将这些值连接到原始数据框的末尾(在删除已经存在的 Result 列之后)以获得如下内容:

ID   A   B   C   Other_val
1    1   0   0   y
2    1   1   0   x
3    0   0   1   abc

但是，我很难过。我似乎无法使用 pd.concat() 来形成上表，因为交叉表的索引方式很奇怪。

有什么帮助吗？

Answer 1

我认为您要查找的是 pd.crosstab([df.ID, df.Other_val], df.Result)，因为您需要对 ID 和 Other_val 进行分组。

In [5]: pd.crosstab([df.ID, df.Other_val], df.Result)
Out[5]: 
Result        A  B  C
ID Other_val         
1  y          1  0  0
2  x          1  1  0
3  abc        0  0  1