Python3 --两列的pandas交叉表

Question

我正在尝试使用本地数据帧中的pd.crosstab创建一个列联表。假设我们问了两个不同组的三个人他们是否喜欢冰激凌的问题，下面是数据框中的结果：

group1    | group2    
------------------
yes       | no    
no        | maybe 
yes       | no    

我希望列联表是这样的：

      | group1     | group2
----------------------------
yes   | 2          | 0
no    | 1          | 2
maybe | 0          | 1 

我玩过熊猫游戏，显然参考了许多不同的资源，包括文档和其他帖子，但我无法理解这一点。有谁有什么想法吗？谢谢!

Answer 1

Pandas有一个crosstab函数可以解决这个问题；首先你必须对数据帧执行melt操作：

box = df.melt()
pd.crosstab(box.value, box.variable)

variable    group1  group2
value       
maybe          0    1
no             1    2
yes            2    0

对于性能，groupby可能会更快，即使它涉及到更多的步骤：

box.groupby(["variable", "value"]).size().unstack("variable", fill_value=0)