Merge pull request matplotlib#30367 from anntzer/st

Support passing xticks/yticks when constructing secondary_axis.

Author: QuLogic

lib/matplotlib/axes/_secondary_axes.py

@@ -1,3 +1,4 @@
+import functools
 import numbers
 
 import numpy as np
@@ -145,10 +146,25 @@ def apply_aspect(self, position=None):
         self._set_lims()
         super().apply_aspect(position)
 
-    @_docstring.copy(Axis.set_ticks)
-    def set_ticks(self, ticks, labels=None, *, minor=False, **kwargs):
-        ret = self._axis.set_ticks(ticks, labels, minor=minor, **kwargs)
-        self.stale = True
+    @functools.wraps(_AxesBase.set_xticks)
+    def set_xticks(self, *args, **kwargs):
+        if self._orientation == "y":
+            raise TypeError("Cannot set xticks on a secondary y-axis")
+        ret = super().set_xticks(*args, **kwargs)
+        self._ticks_set = True
+        return ret
+
+    @functools.wraps(_AxesBase.set_yticks)
+    def set_yticks(self, *args, **kwargs):
+        if self._orientation == "x":
+            raise TypeError("Cannot set yticks on a secondary x-axis")
+        ret = super().set_yticks(*args, **kwargs)
+        self._ticks_set = True
+        return ret
+
+    @functools.wraps(Axis.set_ticks)
+    def set_ticks(self, *args, **kwargs):
+        ret = self._axis.set_ticks(*args, **kwargs)
         self._ticks_set = True
         return ret
 

lib/matplotlib/axes/_secondary_axes.pyi

@@ -29,6 +29,22 @@ class SecondaryAxis(_AxesBase):
         location: Literal["top", "bottom", "right", "left"] | float,
         transform: Transform | None = ...
     ) -> None: ...
+    def set_xticks(
+        self,
+        ticks: ArrayLike,
+        labels: Iterable[str] | None = ...,
+        *,
+        minor: bool = ...,
+        **kwargs
+    ) -> list[Tick]: ...
+    def set_yticks(
+        self,
+        ticks: ArrayLike,
+        labels: Iterable[str] | None = ...,
+        *,
+        minor: bool = ...,
+        **kwargs
+    ) -> list[Tick]: ...
     def set_ticks(
         self,
         ticks: ArrayLike,

lib/matplotlib/tests/test_axes.py

@@ -8193,6 +8193,18 @@ def test_secondary_formatter():
         secax.xaxis.get_major_formatter(), mticker.ScalarFormatter)
 
 
+def test_secondary_init_xticks():
+    fig, ax = plt.subplots()
+    secax = ax.secondary_xaxis(1, xticks=[0, 1])
+    assert isinstance(secax.xaxis.get_major_locator(), mticker.FixedLocator)
+    with pytest.raises(TypeError):
+        secax.set_yticks([0, 1])
+    secax = ax.secondary_yaxis(1, yticks=[0, 1])
+    assert isinstance(secax.yaxis.get_major_locator(), mticker.FixedLocator)
+    with pytest.raises(TypeError):
+        secax.set_xticks([0, 1])
+
+
 def test_secondary_repr():
     fig, ax = plt.subplots()
     secax = ax.secondary_xaxis("top")