add text_auto to px

Author: nicolaskruchten

packages/python/plotly/plotly/express/_chart_types.py

@@ -103,6 +103,7 @@ def density_contour(
     histnorm=None,
     nbinsx=None,
     nbinsy=None,
+    text_auto=False,
     title=None,
     template=None,
     width=None,
@@ -172,6 +173,7 @@ def density_heatmap(
     histnorm=None,
     nbinsx=None,
     nbinsy=None,
+    text_auto=False,
     title=None,
     template=None,
     width=None,
@@ -353,6 +355,7 @@ def bar(
     log_y=False,
     range_x=None,
     range_y=None,
+    text_auto=False,
     title=None,
     template=None,
     width=None,
@@ -454,6 +457,7 @@ def histogram(
     histfunc=None,
     cumulative=None,
     nbins=None,
+    text_auto=False,
     title=None,
     template=None,
     width=None,

packages/python/plotly/plotly/express/_core.py

@@ -1791,6 +1791,16 @@ def infer_config(args, constructor, trace_patch, layout_patch):
     ):
         args["histfunc"] = trace_patch["histfunc"] = "sum"
 
+    if args.get("text_auto", False) is not False:
+        if constructor in [go.Histogram2d, go.Histogram2dContour]:
+            letter = "z"
+        else:
+            letter = "y" if args["orientation"] == "v" else "x"
+        if args["text_auto"] is True:
+            trace_patch["texttemplate"] = "%{" + letter + "}"
+        else:
+            trace_patch["texttemplate"] = "%{" + letter + ":" + args["text_auto"] + "}"
+
     if constructor in [go.Histogram2d, go.Densitymapbox]:
         show_colorbar = True
         trace_patch["coloraxis"] = "coloraxis1"

packages/python/plotly/plotly/express/_doc.py

@@ -591,6 +591,11 @@
         "If `'complementary'`, the CCDF is plotted such that values represent data above the point.",
         "If `'reversed'`, a variant of the CCDF is plotted such that values represent data at or above the point.",
     ],
+    text_auto=[
+        "bool or string (default `False`)",
+        "If `True` or a string, the x or y or z values will be displayed as text, depending on the orientation",
+        "A string like `'.2f'`, it will be interpreted as a `texttemplate` numeric formatting directive.",
+    ],
 )
 
 

packages/python/plotly/plotly/express/_imshow.py

@@ -79,6 +79,7 @@ def imshow(
     binary_backend="auto",
     binary_compression_level=4,
     binary_format="png",
+    text_auto=False,
 ):
     """
     Display an image, i.e. data on a 2D regular raster.
@@ -208,6 +209,10 @@ def imshow(
         since it uses lossless compression, but 'jpg' (lossy) compression can
         result if smaller binary strings for natural images.
 
+    text_auto: bool or str (default `False`)
+        If `True` or a string, single-channel `img` values will be displayed as text",
+        A string like `'.2f'`, it will be interpreted as a `texttemplate` numeric formatting directive.
+
     Returns
     -------
     fig : graph_objects.Figure containing the displayed image
@@ -385,8 +390,21 @@ def imshow(
                 "The length of the x vector must match the length of the second "
                 + "dimension of the img matrix."
             )
+
+        if text_auto is True:
+            texttemplate = "%{z}"
+        elif text_auto is not False:
+            texttemplate = "%{z:" + text_auto + "}"
+
         traces = [
-            go.Heatmap(x=x, y=y, z=img[index_tup], coloraxis="coloraxis1", name=str(i))
+            go.Heatmap(
+                x=x,
+                y=y,
+                z=img[index_tup],
+                coloraxis="coloraxis1",
+                name=str(i),
+                texttemplate=texttemplate,
+            )
             for i, index_tup in enumerate(itertools.product(*iterables))
         ]
         autorange = True if origin == "lower" else "reversed"