comparison utils.py @ 8:4d1bfd404f6a

utils.py(): Added function int_setticks() to set axes labels, fontsize and tick locations and labels
author nikcleju
date Wed, 18 Jan 2012 17:50:04 +0000
parents b4a0b4dfe206
children 7fdf964f4edd
comparison
equal deleted inserted replaced
7:b4a0b4dfe206 8:4d1bfd404f6a
64 # bigmatrix[i*N:i*N+N,j*N+N-border:j*N+N] = bordercolor 64 # bigmatrix[i*N:i*N+N,j*N+N-border:j*N+N] = bordercolor
65 65
66 plt.figure() 66 plt.figure()
67 #plt.imshow(mdict['meanmatrix'][algoname][0,0], cmap=cm.gray, interpolation='nearest',origin='lower') 67 #plt.imshow(mdict['meanmatrix'][algoname][0,0], cmap=cm.gray, interpolation='nearest',origin='lower')
68 plt.imshow(bigmatrix, cmap=cm.gray, norm=mcolors.Normalize(0,1), interpolation='nearest',origin='lower') 68 plt.imshow(bigmatrix, cmap=cm.gray, norm=mcolors.Normalize(0,1), interpolation='nearest',origin='lower')
69 #if algoname == 'GAP': 69 if algoname == 'GAP':
70 int_setticks() 70 int_setticks()
71 #else: 71 else:
72 # plt.gca().get_xaxis().set_visible(False) 72 plt.gca().get_xaxis().set_visible(False)
73 # plt.gca().get_yaxis().set_visible(False) 73 plt.gca().get_yaxis().set_visible(False)
74 74
75 if dosave: 75 if dosave:
76 for ext in saveplotexts: 76 for ext in saveplotexts:
77 plt.savefig(saveplotbase + algoname + '.' + ext, bbox_inches='tight') 77 plt.savefig(saveplotbase + algoname + '.' + ext, bbox_inches='tight')
78 elif mdict['meanmatrix'][algoname][0,0].ndim == 3: 78 elif mdict['meanmatrix'][algoname][0,0].ndim == 3:
155 155
156 return bigmatrix 156 return bigmatrix
157 157
158 def int_setticks(): 158 def int_setticks():
159 159
160 ticks = [10, 94, 179] 160 #ticks = [10, 94, 179]
161 ticklabels = ["0.05", "0.5", "0.95"] 161 #ticklabels = ["0.05", "0.5", "0.95"]
162 ticks = [10, 179]
163 ticklabels = ["0.05", "0.95"]
162 164
163 ax = plt.gca() 165 ax = plt.gca()
164 ax.set_xticks(ticks) 166 ax.set_xticks(ticks)
165 ax.set_xticklabels(ticklabels) 167 ax.set_xticklabels(ticklabels)
166 ax.set_yticks(ticks) 168 ax.set_yticks(ticks)
169 for label in ax.get_xticklabels(): 171 for label in ax.get_xticklabels():
170 label.set_fontsize(42) 172 label.set_fontsize(42)
171 for label in ax.get_yticklabels(): 173 for label in ax.get_yticklabels():
172 label.set_fontsize(42) 174 label.set_fontsize(42)
173 175
174 ax.set_xlabel('\delta') 176 ax.set_xlabel(r'$\delta$', size=60)
177 ax.set_ylabel(r'$\rho$', size=60)
178