Added electrode formatting options to topo plots

scot-dev · Aug 4, 2015 · 5969b1d · 5969b1d
1 parent cee760c
commit 5969b1d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/scot/eegtopo/topoplot.py b/scot/eegtopo/topoplot.py
@@ -146,12 +146,12 @@ def plot_map(self, axes=None, crange=None, offset=(0,0)):
                            extent=(offset[0]-self.interprange, offset[0]+self.interprange,
                                    offset[1]-self.interprange, offset[1]+self.interprange))
 
-    def plot_locations(self, axes=None, offset=(0,0)):
+    def plot_locations(self, axes=None, offset=(0,0), fmt='k.', alpha=1.0):
         if axes is None:
             import matplotlib.pyplot as plot
             axes = plot.gca()
         p2 = array_project_radial_to2d(self.locations) * self.electrodescale + offset
-        axes.plot(p2[:, 0], p2[:, 1], 'k.')
+        axes.plot(p2[:, 0], p2[:, 1], fmt, alpha)
 
     def plot_head(self, axes=None, offset=(0,0)):
         import matplotlib.patches as patches