Correlation 矩阵可视化

最新推荐文章于 2024-07-23 20:15:06 发布

Bill_zhang5

最新推荐文章于 2024-07-23 20:15:06 发布

阅读量2.2k

点赞数

分类专栏：机器学习

本文链接：https://blog.csdn.net/Bill_zhang5/article/details/80228441

版权

机器学习专栏收录该内容

9 篇文章 0 订阅

订阅专栏

import matplotlib.pyplot as plt
import pandas
import numpy as np
data_set ='C:/Users/dell/Desktop/3tables.csv'
names= ['0','1','2']
data_matrix = np.loadtxt(open(data_set, "rb"), delimiter=",", skiprows=0)

correlations =np.corrcoef(data_matrix,rowvar=0)
print(correlations)
# plot correlation matrix
fig = plt.figure()
ax = fig.add_subplot(111)
cax = ax.matshow(correlations, vmin=0.1, vmax=0.3)
fig.colorbar(cax)
ticks = np.arange(0,3,1)
ax.set_xticks(ticks)
ax.set_yticks(ticks)
ax.set_xticklabels(names)
ax.set_yticklabels(names)
plt.show()