Provides a custom plotting style for matplotlib and other libraries building on it like pandas or seaborn. The plotting style is set to aproximately the styles described in our wiki: https://wiki.cpi.imtek.uni-freiburg.de/ScientificWriting
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
67 lines
2.3 KiB
67 lines
2.3 KiB
4 years ago
|
from pathlib import Path
|
||
|
|
|
||
|
|
file_custom_edits = Path("./custom cpi styles.txt")
|
||
|
|
file_defaults = Path("./default.mplstyle")
|
||
|
|
file_cpi_style = Path("./cpi.mplstyle")
|
||
|
|
|
||
|
|
cpi_style_header = """
|
||
|
|
### CPI matplotlib plotting style
|
||
|
|
|
||
|
|
# the plotting styles are set to aproximately the styles described in
|
||
|
|
# https://wiki.cpi.imtek.uni-freiburg.de/ScientificWriting
|
||
|
|
# in the file `140109-scientific-writing-06-figures-and-figure-captions.pptx`
|
||
|
|
|
||
|
|
# usage with matplotlib / pyplot:
|
||
|
|
# > from matplotlib import pyplot
|
||
|
|
# > pyplot.style.use('./cpi.mplstyle')
|
||
|
|
|
||
|
|
# usage with other libraries based on matplotlib, like searborn:
|
||
|
|
# > from matplotlib import pyplot
|
||
|
|
# > import seaborn
|
||
|
|
# > pyplot.style.use('./cpi.mplstyle')
|
||
|
|
#
|
||
|
|
# be aware that the set* functions of seaborn (like set_theme) might
|
||
|
|
# mess with the matplotlib style.
|
||
|
|
#
|
||
|
|
# to use the seaborn.set_palette() function to change line colors seems
|
||
|
|
# to be fine
|
||
|
|
|
||
|
|
# to use this style as your default matplotlib style, have a look at
|
||
|
|
# https://matplotlib.org/stable/tutorials/introductory/customizing.html#customizing-with-matplotlibrc-files
|
||
|
|
|
||
|
|
"""
|
||
|
|
|
||
|
|
|
||
|
|
def get_custom_edits(filepath=file_custom_edits):
|
||
|
|
with filepath.open("r") as fh:
|
||
|
|
lines = (line.strip() for line in fh)
|
||
|
|
not_empty = (line for line in lines if line)
|
||
|
|
with_data = (line for line in not_empty if ":" in line)
|
||
|
|
raw_data = (line.split(":", 1) for line in with_data)
|
||
|
|
data = ((k.strip(), v.strip()) for k, v in raw_data)
|
||
|
|
return dict(sorted(data))
|
||
|
|
|
||
|
|
custom_edits = get_custom_edits(file_custom_edits)
|
||
|
|
key_max_length = 1 + max(len(k) for k in custom_edits)
|
||
|
|
|
||
|
|
|
||
|
|
with open(file_cpi_style, "w") as fw:
|
||
|
|
fw.write(cpi_style_header)
|
||
|
|
|
||
|
|
for key, value in custom_edits.items():
|
||
|
|
line = f"{key:<{key_max_length}}: {value}"
|
||
|
|
fw.write(f"{line}\n")
|
||
|
|
|
||
|
|
fw.write(f"\n\n\n")
|
||
|
|
fw.write(f"### Following are the default matplotlib settings\n")
|
||
|
|
fw.write(f"### Custom styles are commented out\n\n")
|
||
|
|
|
||
|
|
with open(file_defaults, "r") as fr:
|
||
|
|
for line in fr:
|
||
|
|
try:
|
||
|
|
key = line.split(":")[0]
|
||
|
|
if key.strip() in custom_edits:
|
||
|
|
line = f"#{line}"
|
||
|
|
except ValueError():
|
||
|
|
pass
|
||
|
|
fw.write(line)
|