Correlation matrix visualization

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()

Guess you like

Origin http://43.154.161.224:23101/article/api/json?id=325848651&siteId=291194637