Add points return type for scatter

Author: freeman-lab

lightning/types/plots.py

@@ -71,14 +71,23 @@ def clean(x, y, color=None, label=None, value=None, colormap=None, size=None, al
 
     def selected(self):
         """
-        Selected points from scatter plot
+        Selected points from scatter plot as indices
         """
         user_data = self.get_user_data()['settings']
         if 'selected' in user_data.keys():
             return user_data['selected']
         else:
             return []
 
+    def points(self):
+        """
+        Selected points from scatter plot as x,y coordinates
+        """
+        user_data = self.get_user_data()['settings']
+        if 'x' in user_data.keys() and 'y' in user_data.keys():
+            return user_data['x'], user_data['y']
+        else:
+            return []
 
 @viztype
 class Matrix(Base):