Recently I have been combining the two packages to create maps of events happening around the world. The size and colour of the marker on the map can be varied, meaning it’s possible to fit quite a lot of information in one graphic.
However one thing I really struggled with was the legend. If you use different colour points matplotlib makes it easy to add a colour bar, with something like:
c = plt.colorbar(orientation='vertical', shrink = 0.5)
Shrink gives you a quick way of adjusting the size of the bar relative to the graphic.
However I couldn’t find an equivalent command which would give me a legend for the size of the points (something that ggplot does easily). After fiddling with
get_label() for ages, and trying to capture and do something useful with the results of
plt.scatter(), I finally came across this useful post, which basically says that this feature doesn’t really exist and if you want such a legend you have to make it yourself. However, the trick to doing it is quite simple – draw three or four points on your plot with location set to , , (so they won’t actually show up), each one representing a certain size in your scale. These points can then be passed to
plt.legend with some hand written labels. Overall it looks something like this:
l1 = plt.scatter(,, s=10, edgecolors='none')
l2 = plt.scatter(,, s=50, edgecolors='none')
l3 = plt.scatter(,, s=100, edgecolors='none')
l4 = plt.scatter(,, s=200, edgecolors='none')
labels = ["10", "50", "100", "200"]
leg = plt.legend([l1, l2, l3, l4], labels, ncol=4, frameon=True, fontsize=12,
handlelength=2, loc = 8, borderpad = 1.8,
handletextpad=1, title='My Title', scatterpoints = 1)
Well, I still think that should be easier, but at least it works and it also gives you a lot of flexibility with what goes on the legend.