pyplot: change ncols within a legend

matplotlib multiple-columns legend-properties
Tyrax picture Tyrax · Jul 28, 2013 · Viewed 9.9k times · Source

I'm plotting a large data set and some regressions in pyplot. The data is colored according to an additional value. I decided to set the number of columns in the legend to 2.

It looks nice for the data points, but for the regressions, I'd like to go back to ncols=1. It is possible to do this within one legend?

(I know, I could declare two legends but I'd like to avoid this...)

Answer

Jae-Joon Lee picture Jae-Joon Lee · Aug 6, 2013

Legend in matplotlib is a collection of OffsetBox. In principle, you can change "ncol" within a legend by manually rearranging these offsetboxes. Of course, this needs some knowledge of matplotlib internals and will be difficult for normal users. Not sure what would be the best way to expose this. Here is a simple code that creates two legends and combine them into one.

import matplotlib.pyplot as plt
import matplotlib.legend as mlegend

ax = plt.subplot(111)
l1, = ax.plot([1,2,3])

leg1 = ax.legend([l1], ["long lable"], ncol=1)

leg2 = mlegend.Legend(ax, [l1, l1, l1, l1], tuple("1234"), ncol=2)

leg1._legend_box._children.append(leg2._legend_box._children[1])
leg1._legend_box.align="left" # the default layout is 'center'

plt.show()

Related questions

How do you change the size of figures drawn with matplotlib?

How do you change the size of figure drawn with matplotlib?

Read More
Save plot to image file instead of displaying it using Matplotlib

I am writing a quick-and-dirty script to generate plots on the fly. I am using the code below (from Matplotlib documentation) as a starting point: from pylab import figure, axes, pie, title, show # Make a square figure and axes figure(1, …

Read More
How do I set the figure title and axes labels font size in Matplotlib?

I am creating a figure in Matplotlib like this: from matplotlib import pyplot as plt fig = plt.figure() plt.plot(data) fig.suptitle('test title') plt.xlabel('xlabel') plt.ylabel('ylabel') fig.savefig('test.jpg') I want to specify font …

Read More