如何在 Python 中为两个分类变量创建数据透视表？

Question

我的数据看起来像：

Index	Job	Y	Balance
1	A	Yes	1
2	B	No	2
3	A	No	5
4	A	No	0
5	B	Yes	4

我想按以下格式汇总数据，行中有作业，列中有 Y：

	Yes	No
A	1	2
B	1	1

我尝试了以下代码：

pivot = df.pivot_table(index =['job'], columns = ['y'], values = ['balance'], aggfunc ='count')

如果不在 value 参数中使用 balance，我将无法运行 pivot。我如何获得上述结果？

Answer 1

你可以试试这个：

data = {'Index': [1, 2, 3, 4, 5],
'job': ['A', 'B', 'A', 'A', 'B'],
'y': ['Yes', 'No', 'No', 'No', 'Yes'],
'Balance': [1, 2, 5, 0, 4]}
df = pd.DataFrame(data)

pivot = df.groupby(['job', 'y']).size().unstack(fill_value=0)

print(pivot)