Easily set themes that automatically apply to matplotlib, seaborn, and pandas plots
Project description
pyplot-themes
Themes you can see that apply to matplotlib, seaborn, and pandas plots.
This package was inspired by the ggthemes
package in R,
and the code influenced from the seaborn
package in python (specifically rcmod.py
).
Installing
Install from PyPI
pip install pyplot-themes
Or directly from GitHub
pip install git+https://github.com/raybuhr/pyplot-themes.git
Usage
Environment
import sys
sys.version
'3.7.1 (default, Dec 14 2018, 19:28:38) \n[GCC 7.3.0]'
import matplotlib.pyplot as plt
from seaborn import palplot # only used to show off palettes
from string import ascii_uppercase
import numpy as np
def example_scatter_plot(num_cats=6):
for i in range(num_cats):
cat = ascii_uppercase[i]
x = np.random.random(100)
y = np.random.random(100) + i
plt.scatter(x, y, marker='o', label=cat)
plt.legend(loc='best')
def example_bar_plot(num_cats=6):
bar_width = 1 / num_cats + 1
for i in range(num_cats):
cat = ascii_uppercase[i]
x = np.arange(11) + 5 * i
y = np.array([0, 1, 2, 3, 4, 5, 4, 3, 2, 1, 0]) + np.random.random(1)
plt.bar(x, y, label=cat, width=bar_width)
plt.legend(loc='best')
def example_plots(num_cats=6):
example_scatter_plot(num_cats)
plt.show()
example_bar_plot(num_cats)
plt.show()
Default Maplotlib Theme
example_plots()
As you can see, the default theme has good contrast in colors, but leaves a bit to be desired in the sige of the chart (i.e. figure size aka figsize) and font.
Usage
import pyplot_themes as themes
themes.__version__
'0.2.0'
themes.theme_minimal()
This updates the global theme settings for matplotlib with a nice minimal style using colorblind safe colors.
palplot(themes.palettes.Colorblind.colors)
example_plots()
As you can see, our plots are much larger now, have accessible colors, and have some light gridlines to make identifying values a bit easier.
There are a few parameters available in all themes:
- grid: toggles grid lines on/off
- ticks: toggles tick marks on/off
- figsize: sets the default size of plots (you can still change each plot in an ad hoc manner if needed)
- fontsize: sets the default font size to be used
Some themes will allow you to pass in whatever colors you want, others you have to pick a color scheme from available options, some only let you reverse the order of the default color palette, and some don't let you mess with the colors at all. Experiment and find out what you like.
themes.theme_minimal(grid=False, ticks=False, fontsize=18)
example_scatter_plot()
plt.title("Look Mom, no lines!")
Text(0.5, 1.0, 'Look Mom, no lines!')
Themes
themes.theme_dark()
example_plots()
themes.theme_tableau()
example_plots()
palplot(themes.palettes.Solarized.dark)
themes.theme_solarized(scheme="dark")
example_plots()
palplot(themes.palettes.Solarized.light)
themes.theme_solarized(scheme="light")
example_plots()
palplot(themes.palettes.PaulTolColorSchemes.colors)
themes.theme_paul_tol()
example_plots(12)
themes.theme_paul_tol(reverse_colors=True, grid=False)
example_plots(num_cats=12)
palplot(themes.palettes.Few.light)
palplot(themes.palettes.Few.medium)
palplot(themes.palettes.Few.dark)
themes.theme_few(scheme="light")
example_plots()
themes.theme_few(scheme="medium", figsize=[5, 5])
example_scatter_plot()
themes.theme_few(scheme="dark")
example_bar_plot()
themes.theme_ucberkeley(figsize=[10, 5])
example_plots(num_cats=4)
themes.theme_ucberkeley(scheme="all", figsize=[12, 6])
example_plots(num_cats=16)
Themes that come with matplotlib
These next themes actually come with matplotlib and you can use them without the pyplot-themes
package.
The functions here are basically thin wrappers for calling the matplotlib defined styles, but use a bigger figsize by default.
themes.theme_fivethirtyeight()
example_plots()
themes.theme_ggplot2(figsize=[10, 5])
example_plots()
bmh
stands for Bayesian Methods for Hackers
themes.theme_bmh()
example_scatter_plot()
So we also have an alias for the spelled out version to make it easier to discover
themes.theme_bayesian_methods_for_hackers()
example_bar_plot()
While this package provides light and dark solarized themes, matplotlib
comes with a light version as well. This one is a good choice if you want to keep more contrast in the colors of your plots.
themes.theme_solarized_light2()
example_plots()
Modifying Themes
In addition to making it easy to find and call the matplotlib themes, pyplot-themes
also makes it easier to modify them slightly. For example say you want to use the ggplot2
theme, but you want to use the Paul Tol Color Schemes palette with it.
themes.theme_ggplot2(palette=themes.palettes.PaulTolColorSchemes.colors, figsize=[12, 6])
example_bar_plot(num_cats=12)
Or maybe the fivethirtyeight
colors
themes.theme_ggplot2(palette=themes.palettes.FiveThirtyEight.colors)
example_bar_plot()
Resetting to back to matplotlib defaults
Of course, sometimes when you are trying out different themes, you may find you modified a setting that you didn't quite like, but aren't sure what changed. To aid in debugging, we created a function to reset the theme back to what matplotlib starts with. Of course, you may just like the matplotlib defaults and that's ok.
Note: The default settings for matplotlib can be slightly different depending on if you are using in python files (e.g. scripts) vs. in jupyter notebooks using %matplotlib inline
. The reset function assumes you are using a notebook by default, but provides a parameter to toggle that off if you are not:
themes.theme_reset(notebook=False)
themes.theme_reset() # could also use the alias `themes.theme_matplotlib_default()`
example_bar_plot()
Palettes
In addition to the themes above, there are a bunch of color palettes provided. Here are a few to show off.
palplot(themes.palettes.Autumn1.colors)
palplot(themes.palettes.Autumn2.colors)
palplot(themes.palettes.Canyon.colors)
palplot(themes.palettes.Chili.colors)
palplot(themes.palettes.Tomato.colors)
palplot(themes.palettes.Few.medium)
palplot(themes.palettes.FiveThirtyEight.colors)
palplot(themes.palettes.Solarized.light)
palplot(themes.palettes.Solarized.dark)
palplot(themes.palettes.UCBerkeley.primary_colors)
palplot(themes.palettes.UCBerkeley.secondary_colors)
Sequential Palettes
palplot(themes.palettes.Sequential.blues)
palplot(themes.palettes.Sequential.cyans)
palplot(themes.palettes.Sequential.purples)
palplot(themes.palettes.Sequential.greens)
palplot(themes.palettes.Sequential.oranges)
palplot(themes.palettes.Sequential.reds)
Diverging Palettes
palplot(themes.palettes.Diverging.blueorange)
palplot(themes.palettes.Diverging.orangeblue)
palplot(themes.palettes.Diverging.bluepurple)
palplot(themes.palettes.Diverging.purpleblue)
palplot(themes.palettes.Diverging.bluered)
palplot(themes.palettes.Diverging.redblue)
palplot(themes.palettes.Diverging.greenpurple)
palplot(themes.palettes.Diverging.purplegreen)
palplot(themes.palettes.Diverging.greenred)
palplot(themes.palettes.Diverging.redgreen)
Using with Pandas
import pandas as pd
# some made up date
sales = np.random.randint(low=10, high=20, size=30) * [i**2 for i in range(1, 31)]
revenue = np.random.random(30) * sales
months = pd.date_range(start="2010-01-01", periods=30, freq="M")
df = pd.DataFrame({"sales": sales, "revenue": revenue.round(2)}, index=months)
df.head()
.dataframe tbody tr th {
vertical-align: top;
}
.dataframe thead th {
text-align: right;
}
</style>
sales | revenue | |
---|---|---|
2010-01-31 | 12 | 2.76 |
2010-02-28 | 52 | 45.05 |
2010-03-31 | 90 | 11.80 |
2010-04-30 | 208 | 203.93 |
2010-05-31 | 475 | 337.08 |
themes.theme_minimal()
df.plot()
<matplotlib.axes._subplots.AxesSubplot at 0x7fdc5285f2b0>
themes.theme_dark(palette=themes.palettes.Autumn1.colors)
df.plot()
<matplotlib.axes._subplots.AxesSubplot at 0x7fdc5297e668>
Contributing
There are multiple ways you can help out with this project:
- submit a bug report
- submit a feature request
- Fork this git repo, change some code, and submit a Pull Request
- adding documentation or examples counts as changing code
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for pyplot_themes-0.2.2-py2.py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | c7a819c345410e13bb108d73746d8b3778e32513cdc31b2d5c83428b42bc76cc |
|
MD5 | bc273cb8117c29e90867dde1a9d40bc1 |
|
BLAKE2b-256 | f7e1192d14c94aff4a16c5e688a690a1939e0f258e3d51910c3a4dc75837cf53 |