【Python】相关系数矩阵的可视化作图

yebenli

1. #import matplotlib.cm
   
2. import matplotlib.colors as col
   
3. import matplotlib.pyplot as plt
   
4. import numpy as np
   
5.    
   
6. class DrawPcolor(object):
   
7.     def __init__(self):
   
8.         ##self define the colorbar
   
9.         startcolor = '#006400'  # a dark green
   
10.         midcolor = '#ffffff'    # a bright white
    
11.         endcolor = '#ee0000'    # a dark red
    
12.         self.Mycmap = col.LinearSegmentedColormap.from_list('MyColorbar',[startcolor,midcolor,endcolor])#use the "fromList() method
    
13. 
    
14.     def Pcolor(self,*args,**kwargs):
    
15.         #*args is a tuple,**kwargs is a dict;
    
16.         #Here args means the Matrix Corr,kwargs includes the key of the " AddText function"
    
17.         self.fig=plt.figure()
    
18.         self.ax = self.fig.add_subplot(111)   
    
19.         self.Data=args[0];
    
20.         heatmap = self.ax.pcolor(self.Data,cmap=self.Mycmap);#cmap=plt.cm.Reds)
    
21.         self.fig.colorbar(heatmap)
    
22.         # want a more natural, table-like display
    
23.         self.ax.invert_yaxis()
    
24.         self.ax.xaxis.tick_top()
    
25.         self.ax.set_xticks(np.arange(self.Data.shape[0]) + 0.5, minor=False)
    
26.         self.ax.set_yticks(np.arange(self.Data.shape[1]) + 0.5, minor=False)
    
27.         
    
28.         if kwargs['AddText']==True:
    
29.             for y in range(self.Data.shape[1]):
    
30.                 for x in range(self.Data.shape[0]):
    
31.                     self.ax.text(x + 0.5, y + 0.5, '%.2f' % self.Data[y, x],
    
32.                              horizontalalignment='center',
    
33.                              verticalalignment='center',
    
34.                              )
    
35.         self.fig.show()
    
36. 
    
37.     def Set_labelticks(self,*tick_labels):
    
38.         # put the major ticks at the middle of each cell
    
39.         self.ax.set_xticklabels(tick_labels[0],rotation=0, minor=False)
    
40.         self.ax.set_yticklabels(tick_labels[1],rotation=0, minor=False )
    
41.         self.fig.show()
    
42. 
    
43. def Main():
    
44.     data = np.random.rand(5, 4)
    
45.     Corr = np.corrcoef(data)
    
46.     xlabel_ticks=["a",'b','c','d','e']#range(Corr.shape[0]);#
    
47.     ylabel_ticks=["A","B","C",'D','E']#range(Corr.shape[1]);#
    
48.     MyPict=DrawPcolor()
    
49.     MyPict.Pcolor(Corr,3,4,4,AddText=True,b=1) # params 3,4,4,b has no effect on the exec
    
50.     MyPict.Set_labelticks(xlabel_ticks,ylabel_ticks)
    
51. 
    
52. if __name__=='__main__':
    
53.     Main()
    
54. 
    
55.         
    
56.         

复制代码