-
Notifications
You must be signed in to change notification settings - Fork 572
/
theme_gray.py
93 lines (84 loc) · 4.24 KB
/
theme_gray.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
from __future__ import (absolute_import, division, print_function,
unicode_literals)
import matplotlib as mpl
from cycler import cycler
from .theme import theme_base
class theme_gray(theme_base):
"""
Standard theme for ggplot. Gray background w/ white gridlines.
Copied from the the ggplot2 codebase:
https://github.com/hadley/ggplot2/blob/master/R/theme-defaults.r
"""
def __init__(self):
super(theme_gray, self).__init__()
self._rcParams["timezone"] = "UTC"
self._rcParams["lines.linewidth"] = "1.0"
self._rcParams["lines.antialiased"] = "True"
self._rcParams["patch.linewidth"] = "0.5"
self._rcParams["patch.facecolor"] = "348ABD"
self._rcParams["patch.edgecolor"] = "#E5E5E5"
self._rcParams["patch.antialiased"] = "True"
self._rcParams["font.family"] = "sans-serif"
self._rcParams["font.size"] = "12.0"
self._rcParams["font.serif"] = ["Times", "Palatino",
"New Century Schoolbook",
"Bookman", "Computer Modern Roman",
"Times New Roman"]
self._rcParams["font.sans-serif"] = ["Helvetica", "Avant Garde",
"Computer Modern Sans serif",
"Arial"]
self._rcParams["axes.facecolor"] = "#E5E5E5"
self._rcParams["axes.edgecolor"] = "bcbcbc"
self._rcParams["axes.linewidth"] = "1"
self._rcParams["axes.grid"] = "True"
self._rcParams["axes.titlesize"] = "x-large"
self._rcParams["axes.labelsize"] = "large"
self._rcParams["axes.labelcolor"] = "black"
self._rcParams["axes.axisbelow"] = "True"
self._rcParams["axes.prop_cycle"] = cycler('color', ["#333333", "#348ABD", "#7A68A6",
"#A60628", "#467821", "#CF4457", "#188487", "#E24A33"])
self._rcParams["grid.color"] = "white"
self._rcParams["grid.linewidth"] = "1.4"
self._rcParams["grid.linestyle"] = "solid"
self._rcParams["xtick.major.size"] = "0"
self._rcParams["xtick.minor.size"] = "0"
self._rcParams["xtick.major.pad"] = "6"
self._rcParams["xtick.minor.pad"] = "6"
self._rcParams["xtick.color"] = "#7F7F7F"
self._rcParams["xtick.direction"] = "out" # pointing out of axis
self._rcParams["ytick.major.size"] = "0"
self._rcParams["ytick.minor.size"] = "0"
self._rcParams["ytick.major.pad"] = "6"
self._rcParams["ytick.minor.pad"] = "6"
self._rcParams["ytick.color"] = "#7F7F7F"
self._rcParams["ytick.direction"] = "out" # pointing out of axis
self._rcParams["legend.fancybox"] = "True"
self._rcParams["figure.figsize"] = "11, 8"
self._rcParams["figure.facecolor"] = "1.0"
self._rcParams["figure.edgecolor"] = "0.50"
self._rcParams["figure.subplot.hspace"] = "0.5"
# TODO: this slows down everything for some reason
# self._rcParams["text.usetex"] = "True"
def apply_final_touches(self, ax):
'''Styles x,y axes to appear like ggplot2
Must be called after all plot and axis manipulation operations have
been carried out (needs to know final tick spacing)
From: https://github.com/wrobstory/climatic/blob/master/climatic/stylers.py
'''
#Remove axis border
for child in ax.get_children():
if isinstance(child, mpl.spines.Spine):
child.set_alpha(0)
#Restyle the tick lines
for line in ax.get_xticklines() + ax.get_yticklines():
line.set_markersize(5)
line.set_markeredgewidth(1.4)
#Only show bottom left ticks
ax.xaxis.set_ticks_position('bottom')
ax.yaxis.set_ticks_position('left')
#Set minor grid lines
ax.grid(True, 'minor', color='#F2F2F2', linestyle='-', linewidth=0.7)
if not isinstance(ax.xaxis.get_major_locator(), mpl.ticker.LogLocator):
ax.xaxis.set_minor_locator(mpl.ticker.AutoMinorLocator(2))
if not isinstance(ax.yaxis.get_major_locator(), mpl.ticker.LogLocator):
ax.yaxis.set_minor_locator(mpl.ticker.AutoMinorLocator(2))