Add selectedpoints style property for plotly Scatter / Points elements

holoviz · Aug 8, 2019 · 514383e · 514383e
1 parent a92408d
commit 514383e
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 2 deletions.
diff --git a/holoviews/plotting/plotly/chart.py b/holoviews/plotting/plotly/chart.py
@@ -27,9 +27,11 @@ class ScatterPlot(ChartPlot, ColorbarPlot):
                                       allow_None=True, doc="""
       Index of the dimension from which the color will the drawn""")
 
-    style_opts = ['marker', 'color', 'cmap', 'alpha', 'size', 'sizemin']
+    style_opts = [
+        'marker', 'color', 'cmap', 'alpha', 'size', 'sizemin', 'selectedpoints'
+    ]
 
-    _nonvectorized_styles = ['cmap', 'alpha', 'sizemin']
+    _nonvectorized_styles = ['cmap', 'alpha', 'sizemin', 'selectedpoints']
 
     trace_kwargs = {'type': 'scatter', 'mode': 'markers'}
 

diff --git a/holoviews/plotting/plotly/element.py b/holoviews/plotting/plotly/element.py
@@ -215,6 +215,10 @@ def graph_options(self, element, ranges, style):
 
             opts[self._style_key] = {STYLE_ALIASES.get(k, k): v
                                      for k, v in styles.items()}
+
+            # Move selectedpoints from style key back to root
+            if 'selectedpoints' in opts.get(self._style_key, {}):
+                opts['selectedpoints'] = opts[self._style_key].pop('selectedpoints')
         else:
             opts.update({STYLE_ALIASES.get(k, k): v
                          for k, v in style.items() if k != 'cmap'})

diff --git a/holoviews/tests/plotting/plotly/testscatterplot.py b/holoviews/tests/plotting/plotly/testscatterplot.py
@@ -52,3 +52,10 @@ def test_scatter_markers(self):
         state = self._get_plot_state(scatter)
         self.assertEqual(state['data'][0]['marker']['symbol'],
                          np.array(['square', 'circle', 'triangle-up']))
+
+    def test_scatter_selectedpoints(self):
+        scatter = Scatter([
+            (0, 1,), (1, 2), (2, 3)
+        ]).options(selectedpoints=[1, 2])
+        state = self._get_plot_state(scatter)
+        self.assertEqual(state['data'][0]['selectedpoints'], [1, 2])