4 Ways To Improve Your Plotly Graphs
Making high-quality data visualizations is a critical skill for Data Scientists. Learn how to get better at it using Plotly's Python library.
These last weeks I've been working on an application using Dash and Plotly. These tools are great if you want to get something out quickly. But, as usual, there's no magical
make_beautiful_graphs parameter you can set to
True by default.
If you want to have beautiful and customized visualizations in your application, you'll need to spend some time playing around Plotly's extensive list of figures' attributes. I wanted to have something better than the default looks, so I went through Plotly's documentation, old code snippets I had, and Stack Overflow questions.
This is not the first time I found myself doing that. However, this time I decided to keep track of the things I frequently do when making graphs with Plotly. That way, I wouldn't need to read the same documentation or browse the same Stack Overflow questions next time.
In this article I'm compiling a list of things I frequently do when building data visualizations using Plotly. Rest assured, I've been working in data-related positions for a while, so you will not find outrageous things like how to make 3D pie charts. Although these improvements are based on a sample of one, I've frequently seen others applying similar ideas.
I'm focusing on practical and simple improvements that apply to most of the basic charts: scatter plots, line charts, bar charts, and some statistical charts. Here you will find things like removing gridlines and not things like selecting the best colors for your 4D contour plot1.
First, I'll do a brief introduction on how you build graphs using Plotly. Next, I'll provide a list of improvements and the reasoning behind them. Last, I'll give additional recommendations I've found useful when working with Plotly and other plotting libraries.
How to Make a Graph Using Plotly
There are three things you need to know about the inner workings of Plotly:
First, making a graph in Plotly is essentially populating a Python dictionary. This dictionary is usually referred to as figure.
Second, there are two keys in the figure dictionary: layout and data. In layout, you define the looks of your graph like the typography, titles, and axes. In the data key, you set the values and information of the traces you'll be plotting. That could be something like [1, 2, 3] for X, [5, 3, 9] for Y and bar chart type.
So, how do you make a figure?
There are multiple ways to do it. The lowest-level approach is to use Python dictionaries, and the highest-level one is using the Plotly Express interface. I tend to use a mid-level interface called Figure Constructor. It's easier to debug than using Python dictionaries, and it's more flexible than Plotly Express.
The code for making a graph using the Figure Constructor looks as follows:
This is the resulting graph:
For the code snippets listed below, I used the Figure Constructor approach. You may need to adjust the code to make it work for your case if you are using a different interface.
So let's get down to the meat and potatoes.
List of Improvements
Here's the list of things I usually do to improve Plotly graphs:
- #1: Remove gridlines and background color
- #2: Keep consistent colors across graphs
- #3: Use spikelines to compare data points
- #4: Remove floating menu, disable zoom and adjust click behavior
#1: Remove gridlines and background color
Gridlines are lines that cross the chart to show axis divisions. They help the viewer visualize the value represented by an unlabeled data point fast. However, gridlines are not very useful when working with interactive graphs. You can hover over a data point and see its value. So more often than not, I remove gridlines when working with Plotly.
Here's how you can do it:
And this is how it looks:
#2: Keep consistent colors across graphs
When working with categories, there are two things people usually like to do. First, they want to assign some specific colors to each group. For instance, if you are analyzing electoral results in the US, you probably want to use particular blue and red variations to identify the Democratic and Republican parties.
Second, you want this color to remain consistent across all the graphs you do. For example, if you are analyzing some real-world companies, you may want to use their distinctive colors to plot their prices but also when you analyze their returns.
Here's how you can do that using Plotly:
The snippet above allows you to keep consistent colors when working with multiple graphs that share the same categories. The key part is the
COLOR_MAPPER dictionary and its use when adding new traces. This dictionary is the mapping of the categories and colors you'll be using across your charts.
Whenever you add a trace to a graph, you can simply assign the right color to the
marker_color attribute by getting it from the
The resulting graph looks as follows:
#3: Use spike lines to compare data points
A spike line is a vertical or horizontal line that appears when hovering on data. It's useful for comparing values in line charts and scatter plots. This is how you can add those using Plotly:
This is the resulting graph:
#4: Remove floating menu, disable zoom and adjust click behavior
I'm not too fond of the floating menu that Plotly adds to your chart by default. It makes graphs look cool, but I've rarely seen people using it. It has so many options that it's just confusing for someone looking at a graph for the first time. Usually, I remove it.
Also, I like to re-define two other user interaction parameters. I prefer to limit the users' ability to zoom in and change the behavior of clicking on a trace in the legend. In Plotly, by default, if you want to inspect a trace on its own, you have to double-click on the trace, instead of just clicking on it. That's not very intuitive, so I tend to invert that behavior.
This is how you can apply those changes:
This is the resulting graph:
There are three things that I've found useful for learning how to make better data visualizations:
- Get feedback from your audience: This is not always possible. But if you can do it, always prioritize getting input from those who will use your data visualizations. If you are working on a dashboard, the first thing you should do is understand what problem your dashboard solves. Then see users interacting with it. That has the highest ROI for your time.
- Check out Storytelling with Data by Cole Knaflic: It's a great book if you want to level-up your data visualization design skills. It provides a lot of practical advice and compelling use cases.
- Ploty's Figure Reference: Get used to Plotly's Figure Reference and Plotly's documentation. You'll be using it a lot. In general, Plotly's documentation is excellent.
I hope you find these ideas useful. There might be some things that do not resonate with you, or others that you feel are missing. If that's the case, please let me know in the comments below. I'll be happy to update this and add other valuable advice.