为了账号安全,请及时绑定邮箱和手机立即绑定

如何使用以下代码构建cramer-v矩阵?

如何使用以下代码构建cramer-v矩阵?

米琪卡哇伊 2022-01-05 19:47:45
我正在尝试使用 cramers 创建一个热图/相关矩阵。我发现下面的代码可以帮助我解决这个问题,但是当使用 itertools.combinations 时,它不会返回与自身的组合,例如 0,0 1,1 等,因此我的矩阵完全错误,因为当一列与自身进行比较时,对角线应该是 1 但它们是 0。我的 20 列中除了 2 列都是分类的,这就是我使用 cramers 的原因def cramers_corrected_stat(confusion_matrix):    """ calculate Cramers V statistic for categorical-categorical association.        uses correction from Bergsma and Wicher,         Journal of the Korean Statistical Society 42 (2013): 323-328    """    chi2 = ss.chi2_contingency(confusion_matrix)[0]    n = confusion_matrix.sum().sum()    phi2 = chi2/n    r,k = confusion_matrix.shape    phi2corr = max(0, phi2 - ((k-1)*(r-1))/(n-1))        rcorr = r - ((r-1)**2)/(n-1)    kcorr = k - ((k-1)**2)/(n-1)    return np.sqrt(phi2corr / min( (kcorr-1), (rcorr-1))) cols = df.columns.to_list()corrM = np.zeros((len(cols),len(cols)))# there's probably a nice pandas way to do thisfor col1, col2 in itertools.combinations(cols, 2):    idx1, idx2 = cols.index(col1), cols.index(col2)    corrM[idx1, idx2] = cramers_corrected_stat(pd.crosstab(df[col1], df[col2]))    corrM[idx2, idx1] = corrM[idx1, idx2]我该如何解决?
查看完整描述

1 回答

?
人到中年有点甜

TA贡献1895条经验 获得超7个赞

我写了一些这样做的东西:github.com/shakedzy/dython

寻找associationsnominal



查看完整回答
反对 回复 2022-01-05
  • 1 回答
  • 0 关注
  • 189 浏览
慕课专栏
更多

添加回答

举报

0/150
提交
取消
意见反馈 帮助中心 APP下载
官方微信