Add cmap argument for mne.viz.utils.plot_sensors (#11720)

doc/changes/latest.inc

@@ -23,6 +23,7 @@ Current (1.5.dev0)
 
 Enhancements
 ~~~~~~~~~~~~
+- Add ``cmap`` argument for the :func:`mne.viz.plot_sensors` (:gh:`11720` by :newcontrib:`Gennadiy Belonosov`)
 - When failing to locate a file, we now print the full path in quotation marks to help spot accidentally added trailing spaces (:gh:`11718` by `Richard Höchenberger`_)
 
 Bugs

doc/changes/names.inc

@@ -170,6 +170,8 @@
 
 .. _Fu-Te Wong: https://github.com/zuxfoucault
 
+.. _Gennadiy Belonosov: https://github.com/Genuster
+
 .. _Geoff Brookshire: https://github.com/gbrookshire
 
 .. _George O'Neill: https://georgeoneill.github.io

mne/viz/utils.py

@@ -1017,6 +1017,8 @@ def plot_sensors(
     sphere=None,
     pointsize=None,
     linewidth=2,
+    *,
+    cmap=None,
     verbose=None,
 ):
     """Plot sensors positions.
@@ -1074,6 +1076,10 @@ def plot_sensors(
         ``kind='3d'``, or ``25`` otherwise.
     linewidth : float
         The width of the outline. If ``0``, the outline will not be drawn.
+    cmap : str | instance of matplotlib.colors.Colormap | None
+        Colormap for coloring ch_groups. Has effect only when ``ch_groups``
+        is list of list. If None, set to ``matplotlib.rcParams["image.cmap"]``.
+        Defaults to None.
     %(verbose)s
 
     Returns
@@ -1197,10 +1203,9 @@ def plot_sensors(
                 color = np.mean(_rgb(x, y, z), axis=0)
                 color_vals[idx, :3] = color  # mean of spatial color
         else:  # array-like
-            import matplotlib.pyplot as plt
-
+            cmap = _get_cmap(cmap)
             colors = np.linspace(0, 1, len(ch_groups))
-            color_vals = [plt.cm.jet(colors[i]) for i in range(len(ch_groups))]
+            color_vals = [cmap(colors[i]) for i in range(len(ch_groups))]
         colors = np.zeros((len(picks), 4))
         for pick_idx, pick in enumerate(picks):
             for ind, value in enumerate(ch_groups):