python::使用corr不能计算category数据类型的相关系数

   在计算一个Dateframe的六列数据的相关系数时,发现只计算其中两个变量

 

 

 

 

corr = round(dataWOE.corr(),4)
corr 

                               

 

 

 

             查找原因:查看数据类型

             

dataWOE.info()

 

 

                    发现不能计算的数据类型为category,改变为float

dataWOE=dataWOE.astype('float')
dataWOE.info()

 

 

 

corr = round(dataWOE.corr(),4)
corr 

 

 

 

mask = np.zeros_like(corr, dtype=np.bool)
mask[np.triu_indices_from(mask)] = True             
plt.figure(figsize = (5, 5))
cmap = sns.diverging_palette(220, 10, as_cmap=True) 
sns.heatmap(corr, mask=mask, cmap=cmap, center=0, annot =True, cbar_kws={"shrink": .5})
plt.show()

 

posted on 2020-11-24 17:07  西兰花座  阅读(602)  评论(0)    收藏  举报

导航