Add support for overriding x tick with non arithmetic progression values

Author: robertoffmoura

plotly/matplotlylib/mpltools.py

@@ -436,9 +436,9 @@ def prep_ticks(ax, index, ax_type, props):
             tick0 = tickvalues[0]
             dticks = [
                 round(tickvalues[i] - tickvalues[i - 1], 12)
-                for i in range(1, len(tickvalues) - 1)
+                for i in range(1, len(tickvalues))
             ]
-            if all([dticks[i] == dticks[i - 1] for i in range(1, len(dticks) - 1)]):
+            if all([dticks[i] == dticks[i - 1] for i in range(1, len(dticks))]):
                 dtick = tickvalues[1] - tickvalues[0]
             else:
                 warnings.warn(
@@ -448,6 +448,8 @@ def prep_ticks(ax, index, ax_type, props):
                 raise TypeError
         except (IndexError, TypeError):
             axis_dict["nticks"] = props["axes"][index]["nticks"]
+            if props["axes"][index]["tickvalues"] is not None:
+                axis_dict["tickvals"] = props["axes"][index]["tickvalues"]
         else:
             axis_dict["tick0"] = tick0
             axis_dict["dtick"] = dtick
@@ -496,6 +498,15 @@ def prep_ticks(ax, index, ax_type, props):
 
     if formatter == "LogFormatterMathtext":
         axis_dict["exponentformat"] = "e"
+    elif (
+        formatter == "FuncFormatter" and props["axes"][index]["tickformat"] is not None
+    ):
+        to_remove = ["dticktickmode"]
+        for key in to_remove:
+            if key in axis_dict:
+                axis_dict.pop(key)
+        axis_dict["ticktext"] = props["axes"][index]["tickformat"]
+        axis_dict["tickvals"] = props["axes"][index]["tickvalues"]
     return axis_dict
 
 

plotly/matplotlylib/tests/test_renderer.py

@@ -84,3 +84,28 @@ def test_multiple_traces_native_legend():
     assert plotly_fig.data[0].mode == "lines"
     assert plotly_fig.data[1].mode == "markers"
     assert plotly_fig.data[2].mode == "lines+markers"
+
+
+
+def test_non_arithmetic_progression_xtickvals():
+    xticks = [0.01, 0.53, 0.75]
+    plt.figure()
+    plt.plot([0, 1], [0, 1])
+    plt.xticks(xticks)
+
+    plotly_fig = tls.mpl_to_plotly(plt.gcf())
+
+    assert plotly_fig.layout.xaxis.tickvals == tuple(xticks)
+
+
+def test_non_arithmetic_progression_xticktext():
+    xtickvals = [0.01, 0.53, 0.75]
+    xticktext = ["Baseline", "param = 1", "param = 2"]
+    plt.figure()
+    plt.plot([0, 1], [0, 1])
+    plt.xticks(xtickvals, xticktext)
+
+    plotly_fig = tls.mpl_to_plotly(plt.gcf())
+
+    assert plotly_fig.layout.xaxis.tickvals == tuple(xtickvals)
+    assert plotly_fig.layout.xaxis.ticktext == tuple(xticktext)