ENH: heatmap drawing as function #5

src/heatmap.py

@@ -20,7 +20,7 @@ def onclick(event):
     print('Indices and content of current cell [{},{}]: {}'.format(idx,idy,data[idy+1,idx]))
 
 
-if __name__=='__main__':
+def draw_heatmap():
 	C = np.loadtxt('../input/colormap.txt')
 	cm = mpc.ListedColormap(C/255.0)
 	'''data = np.array([[1, 2, 3, 4, 5],
@@ -81,4 +81,5 @@ if __name__=='__main__':
 	pp.yticks(rotation=0)
 	pp.xlabel('Frequency [GHz (uncore)]')
 	pp.ylabel('Frequency [GHz (core)]')
-	pp.show()
\ No newline at end of file
+	pp.show()
+