python - 如何使用 pandas 制作自定义混淆矩阵

Question

通过在我的数据帧上使用.value_counts()，我得到了一系列如下:

示例数据:

In_A_Not_In_B    116
In_C_Not_In_B    104
In_A_Not_In_C    100
In_C_Not_In_A    85

我需要根据上述数据创建一个矩阵，如下所示。索引名称必须从列值中获取。

预期输出:

      NOT IN
      A  B  C
   A NA 116 100
IN C 85 104 NA

到目前为止，我已经尝试过 .pivot_table()、.MultiIndex() 但没有成功。

Answer 1

假设您的 value_counts 结果存储在变量 df 中，那么:

df = df.reset_index()
df['In'] = df['index'].str[3]
df['Not In'] = df['index'].str[-1]
df.pivot(index='In', columns='Not In', values=0)
# Not In     A      B      C
# In                        
# A        NaN  116.0  100.0
# C       85.0  104.0    NaN