Clever Geek Handbook

Matplotlib: how to customize the zorder of the second legend?

Here's an example that reproduces my problem:

import matplotlib.pyplot as plt
import numpy as np

data1,data2,data3,data4 = np.random.random(100),np.random.random(100),np.random.random(100),np.random.random(100)

fig,ax = plt.subplots()

ax.plot(data1)
ax.plot(data2)
ax.plot(data3)

ax2 = ax.twinx()
ax2.plot(data4)

plt.grid('on')
ax.legend(['1','2','3'], loc='center')
ax2.legend(['4'], loc=1)

enter image description here

How can I make the center legend draw over the lines?

+3


source to share


1 answer


To get exactly what you asked for, try the following. Note. I modified my code to define labels when plotting, as well as colors so you don't get a repeating blue line.

import matplotlib.pyplot as plt
import numpy as np

data1,data2,data3,data4 = (np.random.random(100),
                           np.random.random(100),
                           np.random.random(100),
                           np.random.random(100))

fig,ax = plt.subplots()

ax.plot(data1, label="1", color="k")
ax.plot(data2, label="2", color="r")
ax.plot(data3, label="3", color="g")

ax2 = ax.twinx()
ax2.plot(data4, label="4", color="b")

# First get the handles and labels from the axes
handles1, labels1 = ax.get_legend_handles_labels()
handles2, labels2 = ax2.get_legend_handles_labels()

# Add the first legend to the second axis so it displaysys 'on top'
first_legend = plt.legend(handles1, labels1, loc='center')
ax2.add_artist(first_legend)

# Add the second legend as usual
ax2.legend(handles2, labels2)

plt.show()

enter image description here



Now I'll add that it would be clearer if you only used one legend adding all the lines to it. This is described in this SO post and in the above code can easily be achieved with

ax2.legend(handles1+handles2, labels1+labels2)

But obviously you may have your own reasons for wanting two legends.

+4


source






More articles:


All Articles