use axis color labels

Author: frutti93

.github/workflows/ci.yml

@@ -23,7 +23,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: [3.6, 3.7, 3.8, 3.9]
+        python-version: ["3.7", "3.8", "3.9", "3.10-dev"]
     steps:
       - uses: actions/setup-python@v2
         with:

.gitignore

@@ -10,3 +10,5 @@ build/
 *.egg-info/
 .pytest_cache/
 .tox/
+mytikz.tex
+test.tex

.pre-commit-config.yaml

@@ -5,12 +5,12 @@ repos:
       - id: isort
 
   - repo: https://github.com/psf/black
-    rev: 21.8b0
+    rev: 21.9b0
     hooks:
       - id: black
         language_version: python3
 
   - repo: https://github.com/PyCQA/flake8
-    rev: 3.9.2
+    rev: 4.0.1
     hooks:
       - id: flake8

README.md

@@ -26,10 +26,9 @@ figures like
 
 for native inclusion into LaTeX or ConTeXt documents.
 
-The output of tikzplotlib is in
-[PGFPlots](https://github.com/pgf-tikz/pgfplots/), a TeX library that sits on
-top of [PGF/TikZ](https://en.wikipedia.org/wiki/PGF/TikZ) and describes graphs in terms
-of axes, data etc. Consequently, the output of tikzplotlib
+The output of tikzplotlib is in [PGFPlots](https://github.com/pgf-tikz/pgfplots/), a TeX
+library that sits on top of [PGF/TikZ](https://en.wikipedia.org/wiki/PGF/TikZ) and
+describes graphs in terms of axes, data etc. Consequently, the output of tikzplotlib
 
 -   retains more information,
 -   can be more easily understood, and
@@ -58,14 +57,17 @@ import tikzplotlib
 
 tikzplotlib.save("test.tex")
 ```
+
 <!--close the figure and reset defaults
 <!--pytest-codeblocks:cont-->
+
 ```python
 import matplotlib as mpl
 
 plt.close()
 mpl.rcParams.update(mpl.rcParamsDefault)
 ```
+
 -->
 (see above) gives
 
@@ -147,8 +149,8 @@ to install.
 
     to store the TikZ file as `mytikz.tex`.
 
-3. Add the contents of `mytikz.tex` into your TeX source code. A convenient way of
-   doing so is via
+3. Add the contents of `mytikz.tex` into your TeX source code. A convenient way of doing
+   so is via
 
     ```latex
     \input{/path/to/mytikz.tex}
@@ -189,7 +191,8 @@ to install.
     tikzplotlib.Flavors.context.preamble()
     ```
 
-4. Optional: clean up the figure before exporting to tikz using the `clean_figure` command.
+4. [Optional] Clean up the figure before exporting to tikz using the `clean_figure`
+   command.
 
     ```python
     import matplotlib.pyplot as plt
@@ -212,16 +215,16 @@ to install.
 
 ### Contributing
 
-If you experience bugs, would like to contribute, have nice examples of what
-tikzplotlib can do, or if you are just looking for more information, then please
-visit [tikzplotlib's GitHub page](https://github.com/nschloe/tikzplotlib).
+If you experience bugs, would like to contribute, have nice examples of what tikzplotlib
+can do, or if you are just looking for more information, then please visit
+[tikzplotlib's GitHub page](https://github.com/nschloe/tikzplotlib).
 
 ### Testing
 
 tikzplotlib has automatic unit testing to make sure that the software doesn't
 accidentally get worse over time. In `test/`, a number of test cases are specified.
-Those run through tikzplotlib and compare the output with a previously stored
-reference TeX file.
+Those run through tikzplotlib and compare the output with a previously stored reference
+TeX file.
 
 To run the tests, just check out this repository and type
 
@@ -231,4 +234,5 @@ pytest
 
 ### License
 
-tikzplotlib is published under the [MIT license](https://en.wikipedia.org/wiki/MIT_License).
+tikzplotlib is published under the [MIT
+license](https://en.wikipedia.org/wiki/MIT_License).

justfile

@@ -7,7 +7,7 @@ default:
 
 tag:
 	@if [ "$(git rev-parse --abbrev-ref HEAD)" != "main" ]; then exit 1; fi
-	curl -H "Authorization: token `cat ~/.github-access-token`" -d '{"tag_name": "{{version}}"}' https://api.github.com/repos/nschloe/{{name}}/releases
+	curl -H "Authorization: token `cat ~/.github-access-token`" -d '{"tag_name": "v{{version}}"}' https://api.github.com/repos/nschloe/{{name}}/releases
 
 upload: clean
 	@if [ "$(git rev-parse --abbrev-ref HEAD)" != "main" ]; then exit 1; fi

setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = tikzplotlib
-version = 0.9.12
+version = 0.9.14
 author = Nico Schlömer
 author_email = [email protected]
 description = Convert matplotlib figures into TikZ/PGFPlots
@@ -14,16 +14,16 @@ long_description_content_type = text/markdown
 license = MIT
 license_file = LICENSE
 classifiers =
-    Development Status :: 4 - Beta
+    Development Status :: 5 - Production/Stable
     Framework :: Matplotlib
     License :: OSI Approved :: MIT License
     Operating System :: OS Independent
     Programming Language :: Python
     Programming Language :: Python :: 3
-    Programming Language :: Python :: 3.6
     Programming Language :: Python :: 3.7
     Programming Language :: Python :: 3.8
     Programming Language :: Python :: 3.9
+    Programming Language :: Python :: 3.10
     Topic :: Multimedia :: Graphics :: Graphics Conversion
     Topic :: Scientific/Engineering :: Visualization
 keywords =
@@ -41,7 +41,7 @@ install_requires =
     matplotlib >= 1.4.0
     numpy
     Pillow
-python_requires = >=3.6
+python_requires = >=3.7
 
 [options.packages.find]
 where=src

src/tikzplotlib/_axes.py

@@ -49,14 +49,30 @@ def __init__(self, data, obj):  # noqa: C901
         xlabel = obj.get_xlabel()
         if xlabel:
             xlabel = _common_texification(xlabel)
-            self.axis_options.append(f"xlabel={{{xlabel}}}")
+
+            labelcolor = obj.xaxis.label.get_c()
+
+            if labelcolor != "black":
+                data, col, _ = _color.mpl_color2xcolor(data, labelcolor)
+                self.axis_options.append(f"xlabel=\\textcolor{{{col}}}{{{xlabel}}}")
+            else:
+                self.axis_options.append(f"xlabel={{{xlabel}}}")
+
             xrotation = obj.xaxis.get_label().get_rotation()
             if xrotation != 0:
                 self.axis_options.append(f"xlabel style={{rotate={xrotation - 90}}}")
+
         ylabel = obj.get_ylabel()
         if ylabel:
             ylabel = _common_texification(ylabel)
-            self.axis_options.append(f"ylabel={{{ylabel}}}")
+
+            labelcolor = obj.yaxis.label.get_c()
+            if labelcolor != "black":
+                data, col, _ = _color.mpl_color2xcolor(data, labelcolor)
+                self.axis_options.append(f"ylabel=\\textcolor{{{col}}}{{{ylabel}}}")
+            else:
+                self.axis_options.append(f"ylabel={{{ylabel}}}")
+
             yrotation = obj.yaxis.get_label().get_rotation()
             if yrotation != 90:
                 self.axis_options.append(f"ylabel style={{rotate={yrotation - 90}}}")

src/tikzplotlib/_cleanfigure.py

@@ -752,7 +752,9 @@ def _move_points_closer(xLim, yLim, data):
 
     dataInsert = np.array([[]])
     if not _isempty(id_replace):
-        raise NotImplementedError
+        raise NotImplementedError(
+            "There is data outside of the box. Don't know how to handle during cleaning. Please check if x/ylim is to tight"
+        )
     data = _insert_data(data, id_replace, dataInsert)
     if _isempty(id_replace):
         return data

src/tikzplotlib/_save.py

@@ -1,8 +1,9 @@
+from __future__ import annotations
+
 import enum
 import pathlib
 import tempfile
 import warnings
-from typing import List, Optional, Set, Union
 
 import matplotlib as mpl
 import matplotlib.pyplot as plt
@@ -17,22 +18,22 @@
 
 def get_tikz_code(
     figure="gcf",
-    filepath: Optional[Union[str, pathlib.Path]] = None,
-    axis_width: Optional[str] = None,
-    axis_height: Optional[str] = None,
+    filepath: str | pathlib.Path | None = None,
+    axis_width: str | None = None,
+    axis_height: str | None = None,
     textsize: float = 10.0,
-    tex_relative_path_to_data: Optional[str] = None,
+    tex_relative_path_to_data: str | None = None,
     externalize_tables: bool = False,
     override_externals: bool = False,
-    externals_search_path: Optional[str] = None,
+    externals_search_path: str | None = None,
     strict: bool = False,
     wrap: bool = True,
     add_axis_environment: bool = True,
-    extra_axis_parameters: Optional[Union[List, Set]] = None,
+    extra_axis_parameters: list | set | None = None,
     extra_groupstyle_parameters: dict = {},
-    extra_tikzpicture_parameters: Optional[Union[List, Set]] = None,
-    extra_lines_start: Optional[Union[List, Set]] = None,
-    dpi: Optional[int] = None,
+    extra_tikzpicture_parameters: list | set | None = None,
+    extra_lines_start: list | set | None = None,
+    dpi: int | None = None,
     show_info: bool = False,
     include_disclaimer: bool = True,
     standalone: bool = False,
@@ -246,9 +247,7 @@ def get_tikz_code(
     return code
 
 
-def save(
-    filepath: Union[str, pathlib.Path], *args, encoding: Optional[str] = None, **kwargs
-):
+def save(filepath: str | pathlib.Path, *args, encoding: str | None = None, **kwargs):
     """Same as `get_tikz_code()`, but actually saves the code to a file.
 
     :param filepath: The file to which the TikZ output will be written.

tests/test_basic_sin_reference.tex

@@ -20,12 +20,12 @@
 tick pos=left,
 title={Simple plot \(\displaystyle \frac{\alpha}{2}\)},
 x grid style={white},
-xlabel={time(s)},
+xlabel=\textcolor{white!33.333333!black}{time(s)},
 xmajorgrids,
 xmin=-0.095, xmax=1.995,
 xtick style={color=white!33.333333!black},
 y grid style={white},
-ylabel={Voltage (mV)},
+ylabel=\textcolor{white!33.333333!black}{Voltage (mV)},
 ymajorgrids,
 ymin=-1.1, ymax=1.1,
 ytick style={color=white!33.333333!black}