A matplotlib extension library for making tree dot plots, strip plots or dot charts in Python (seaborn compatible)
pip install dotplotlib .dotchart returns x and y lists that can be inputted straight into matplotlib or seaborn scatterplots.
from dotplotlib import dotchart import matplotlib.pyplot as plt data = {'size': [1, 2, 2, 3, 3, 3, 4]} # Generate dot chart data x, y = dotchart(data['size']) # Plot plt.scatter(x, y) plt.show()Pass the data you would like to color by to the color_by= argument.
Returns an extra list c that should be passed into the c= parameter if using matplotlib or hue= if using seaborn.
from dotplotlib import dotchart import matplotlib.pyplot as plt data = {'size': [1, 2, 2, 3, 3, 3, 4], 'rating': [3, 2, 5, 4, 3, 6, 4]} # Generate dot chart data with color mapping x, y, c = dotchart(data['size'], color_by=data['rating']) # Plot with color mapping plt.scatter(x, y, c=c, cmap='viridis') plt.colorbar() plt.xlabel('Size') plt.ylabel('Number') plt.title('Mushroom Size Count Colored by Rating') plt.show()Instead of just giving you x, y lists to make the plot yourself, make_dotplot() actually generates the plot.
from dotplotlib import make_dotchart df = {'size': [1, 2, 2, 3, 3, 3, 4], 'rating': [3, 2, 5, 4, 3, 6, 7]} # Create a dot chart with optional arguments (only the first one is mandatory) make_dotchart(df['size'], color_by=df['rating'], # list to color by reverse=False, # inverts the color mapping theme='gnuplot2', # scroll down to see all themes colorbar=True, xlabel='Sizes', ylabel='Size Count', title='Mushroom Sizes Colored by Rating', dot_size=40):If plotting inline, use the default .dotchart() to obtain x and y lists, and then adjust as necessary with one of the following:
plt.figure(figsize=(12,6)) # or plt.figure().set_figwidth(12) # or plt.figure().set_figheight(12)
- Generate strip plots/dot charts by exploiting
matplotlib/seabornscatterplots - Supports any cmap color profile
- The data can be automatically sorted for better visualization, especially when using color mapping.
- Accepts both list and pandas.Series as input data.
- Set custom labels, titles, and dot sizes for your charts.
- Works with Jupyter Notebook
Anyone is welcome to raise a PR!