学习笔记学习计算皮尔相关逊系数,并用热力图展示出来。

       由于是自我练习的笔记,所以这里先通过Pandas随机生成一部分时序数据,然后再调用corr()函数来计算皮尔逊相关系数,并把计算结果先展示输出出来,最后通过热力图的方式把计算结果展现出来。

   下面是开发的具体过程:

1、首先导入需要的算法包

import seaborn as sns
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt

2、生成数据(由于是练习,所以这里的数据是随机生成的时序数据,其中索引为时间)

dates = pd.date_range('20220101', periods=15)#生成连续15天的时间序列
df = pd.DataFrame(np.random.randn(15,4), index=dates, columns=list('ABCD'))
print(df)

输出:

3、通过corr()函数来计算df数据中的两两元素的皮尔相关逊系数

a=df.corr()
print('皮尔逊系数')
print(a)

计算结果如下: 

 

4、通过计算出来的皮尔逊相关系数进行画热力图,用热力图表示相关性。

sns.heatmap(df.corr(method='pearson'),linewidths=0.1,vmax=1.0, square=True,linecolor='white', annot=True)
plt.title('皮尔逊热力图')
plt.show()

输出:

 完整版代码如下(本地运行无误,可以使用):

 #coding=utf-8
import seaborn as sns
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
dates = pd.date_range('20220101', periods=15)#生成连续六天的时间序列
#随机生成数据
df = pd.DataFrame(np.random.randn(15,4), index=dates, columns=list('ABCD'))
print(df)
a=df.corr()
print('皮尔逊系数')
print(a)
plt.rcParams['font.sans-serif'] = ['SimHei']
plt.rcParams['axes.unicode_minus'] = False
sns.heatmap(df.corr(method='pearson'),linewidths=0.1,vmax=1.0, square=True,linecolor='white', annot=True)
plt.title('皮尔逊热力图')
plt.show()

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐