plotnine and seaborn

Author: gordonwoodhull

quarto/__init__.py

@@ -6,8 +6,10 @@
 from quarto.quarto import path
 from quarto.render import render
 from quarto.metadata import metadata
-from quarto.theme import plotly_theme_colors, plotly_theme_brand, \
-    altair_theme_colors, altair_theme_brand, \
+from quarto.theme import altair_theme_colors, altair_theme_brand, \
     bokeh_theme_colors, bokeh_theme_brand, \
     great_tables_theme_colors, great_tables_theme_brand, \
-    matplotlib_theme_colors, matplotlib_theme_brand
+    matplotlib_theme_colors, matplotlib_theme_brand, \
+    plotnine_theme_colors, plotnine_theme_brand, \
+    plotly_theme_colors, plotly_theme_brand, \
+    seaborn_theme_colors, seaborn_theme_brand

quarto/theme.py

@@ -1,19 +1,6 @@
 import yaml
 from functools import partial
 
-def plotly_theme_colors(bg, fg):
-  import plotly.graph_objects as go
-  return go.layout.Template({'layout': {
-    'paper_bgcolor': bg,
-    'plot_bgcolor': bg,
-    'font_color': fg,
-  }})
-
-def plotly_theme_brand(brand_yml):
-  brand = yaml.safe_load(open(brand_yml).read())
-  return plotly_theme_colors(brand["color"]["background"], brand["color"]["foreground"])
-
-
 def altair_theme_colors(bg, fg):
   return {
     'config': {
@@ -99,3 +86,48 @@ def matplotlib_theme_brand(brand_yml):
     brand["color"]["foreground"],
     brand["color"]["primary"],
   )
+
+
+def plotnine_theme_colors(bg, fg):
+  from plotnine import theme, element_rect, element_text
+  return theme(
+    plot_background=element_rect(fill=bg, size=0),
+    text=element_text(color=fg)
+  )
+
+def plotnine_theme_brand(brand_yml):
+  brand = yaml.safe_load(open(brand_yml).read())
+  return plotnine_theme_colors(brand["color"]["background"], brand["color"]["foreground"])
+
+
+def plotly_theme_colors(bg, fg):
+  import plotly.graph_objects as go
+  return go.layout.Template({'layout': {
+    'paper_bgcolor': bg,
+    'plot_bgcolor': bg,
+    'font_color': fg,
+  }})
+
+def plotly_theme_brand(brand_yml):
+  brand = yaml.safe_load(open(brand_yml).read())
+  return plotly_theme_colors(brand["color"]["background"], brand["color"]["foreground"])
+
+
+def seaborn_theme_colors(bg, fg):
+  # seaborn accepts matplotlib rcparams
+  return {
+      "axes.facecolor": bg,
+      "axes.edgecolor": fg,
+      "axes.labelcolor": fg,
+      "axes.titlecolor": fg,
+      "figure.facecolor": bg,
+      "figure.edgecolor": fg,
+      "text.color": fg,
+      "xtick.color": fg,
+      "ytick.color": fg,
+      "grid.color": fg,
+  }
+
+def seaborn_theme_brand(brand_yml):
+    brand = yaml.safe_load(open(brand_yml).read())
+    return seaborn_theme_colors(brand["color"]["background"], brand["color"]["foreground"])