python - 如何合并 pandas 中交叉表的类别？

Question

假设我的 pandas 数据框有 3 个变量 X 类别:[A, B, C] 和 2 个变量 Y 类别:[D，E]。我想对此进行交叉制表，例如:

+--------+----------------------+-----+
|  X/Y   |          D           |  E  |
+--------+----------------------+-----+
| A or B | count(X=A or B, Y=D) | ... |
| C      | count(X=C),Y=D)      | ... |
+--------+----------------------+-----+

Answer 1

这是您要找的吗？

import pandas as pd
import numpy as np
x = np.random.choice(['A', 'B', 'C'], size=10)
y = np.random.choice(['D', 'E'], size=10)
df = pd.DataFrame({'X':x, 'Y':y})
df.head()

输出:

    X   Y
0   A   D
1   B   D
2   B   E
3   B   D
4   A   E

数据框修改:

df['X'] = df['X'].apply(lambda x: 'A or B' if x == 'A' or x == 'B' else x)

交叉表应用程序:

pd.crosstab(df.X, df.Y)

输出:

Y       D   E
X       
A or B  1   3
C       4   2