add custom annotations per facet

Author: trevismd

CHANGELOG.md

@@ -1,4 +1,9 @@
 ## v0.6
+
+### v0.6.1
+- Add option to give facet-specific annotation parameters, such as to
+  set custom annotations
+
 ### v0.6.0
 #### Features
 - Add option to skip annotation of non-significant results 

statannotations/Annotator.py

@@ -1,5 +1,5 @@
 import warnings
-from typing import List, Optional, Union
+from typing import List, Optional, Union, Iterator, Dict, Any
 
 import matplotlib.pyplot as plt
 import matplotlib.transforms as mtransforms
@@ -816,12 +816,16 @@ def _maybe_warn_about_configuration(self):
 
     def plot_and_annotate_facets(
             self, plot: str, plot_params: dict, configuration: dict,
-            annotation_func: str, *args, annotation_params: dict = None,
+            annotation_func: str, *args,
+            annotation_params: dict = None,
+            annotation_params_each: Optional[
+                Iterator[Dict[str, Any]]] = None,
             ax_op_before: List[Union[str, Optional[list],
                                      Optional[dict]]] = None,
             ax_op_after: List[Union[str, Optional[list],
                                     Optional[dict]]] = None,
-            annotate_params: dict = None, **kwargs):
+            annotate_params: dict = None,
+            **kwargs):
         """
         Plots using seaborn and annotates in a single call, to be used within
         a `FacetGrid`.
@@ -836,6 +840,8 @@ def plot_and_annotate_facets(
             * 'set_pvalues'
             * 'apply_test'
         :param annotation_params: parameters for the annotation function
+        :param annotation_params_each: parameters for the annotation function to
+            iterate on to find values for each facet.
         :param ax_op_before: list of [func_name, args, kwargs] to apply on `ax`
             before annotating
         :param ax_op_after: list of [func_name, args, kwargs] to apply on `ax`
@@ -846,14 +852,19 @@ def plot_and_annotate_facets(
         """
         annotate_params = empty_dict_if_none(annotate_params)
         annotation_params = empty_dict_if_none(annotation_params)
+        annotation_params_this = (
+            next(annotation_params_each, {})
+            if annotation_params_each is not None
+            else {}
+        )
 
         ax = getattr(sns, plot)(*args, **plot_params, **kwargs)
 
         _apply_ax_operations(ax, ax_op_before)
 
         self.new_plot(ax, plot=plot, **plot_params, data=kwargs['data'])
         self.configure(**configuration)
-        getattr(self, annotation_func)(**annotation_params)
+        getattr(self, annotation_func)(annotation_params_this, **annotation_params)
         self.annotate(**annotate_params)
 
         _apply_ax_operations(ax, ax_op_after)