Added support for lists of string literals as arbitrary data tooltips

root · root · commit 82a2ab392579 · 2023-05-07T08:10:09.000-04:00
diff --git a/lib/matplotlib/backend_bases.py b/lib/matplotlib/backend_bases.py
@@ -3029,6 +3029,34 @@ def mouse_move(self, event):
                                                     str(newY) +
                                                     "         Original coords: "
                                                     ))
+                        elif type(hover) == list:
+                            import matplotlib.pyplot as plt
+                            #get number of data points first
+                            lines = plt.gca().get_lines()
+                            num_of_points =0
+                            for line in lines:
+                                num_of_points+=1
+                            if num_of_points >= len(hover):
+                                raise ValueError("""Number of data points 
+                                does not match up woth number of labels""")
+                            else:
+                                mouse_x = event.xdata
+                                mouse_y = event.ydata
+                                for line in lines:
+                                    x_data = line.get_xdata()
+                                    y_data = line.get_ydata()
+                                    for i in range(len(x_data)):
+                                        # calculate distance between cursor position and data point
+                                        distance = ((event.xdata - x_data[i])**2 + (event.ydata - y_data[i])**2)**0.5
+                                        if distance < 0.05:  # modify this threshold as needed
+                                            (self.set_hover_message("Data Label: "+ hover[i] +
+                                            "         Original coords: "
+                                            ))
+                          
+
+
+
+
                         else:
                             self.set_hover_message(self._mouse_event_to_message(event))
                 else:
diff --git a/lib/matplotlib/tests/test_toolkit.py b/lib/matplotlib/tests/test_toolkit.py
@@ -22,3 +22,15 @@ def user_defined_function(event):
 
 ax.plot(rand(100), 'o', hover=user_defined_function)
 plt.show()
+
+
+
+############################################################################
+
+#Alternative test for testing out string literals as tooltips
+
+# fig, ax = plt.subplots()
+# plt.ylabel('some numbers')
+
+# ax.plot(rand(3), 'o', hover=['London','Paris','Barcelona'])
+# plt.show()
