Add a top dosctring and perform small cleaning/fixing ops to mri_with_eeg.py

afvincent · afvincent · commit db3d206a4e32 · 2016-10-02T11:27:51.000+02:00
diff --git a/examples/pylab_examples/mri_demo.py b/examples/pylab_examples/mri_demo.py
@@ -1,11 +1,14 @@
-from __future__ import print_function
+"""Displays an MRI image."""
+
 import matplotlib.pyplot as plt
 import matplotlib.cbook as cbook
 import numpy as np
-# data are 256x256 16 bit integers
+
+# Data are 256x256 16 bit integers
 dfile = cbook.get_sample_data('s1045.ima.gz')
 im = np.fromstring(dfile.read(), np.uint16).astype(float)
-im.shape = 256, 256
+im.shape = (256, 256)
+dfile.close()
 
 plt.imshow(im, cmap=plt.cm.gray)
 plt.axis('off')
