Change main plot legend label text


So far I have been able to label the subplots just fine but I’m having an issue with the main one.

Here’s the relevant part of my code:

data_BS_P = data[channels[0]]
data_BS_R = data[channels[1]]
data_BS_Y = data[channels[2]]
plot_BS_P = data_BS_P.plot() #data_BS_P is a pandas dataframe
axBS = plot_BS_P.gca()
axBS.plot(data_BS_R, label='Roll')
axBS.plot(data_BS_Y, label='Yaw')
axBS.set_ylabel('Amplitude (urad)')
axBS.legend(loc='upper center', bbox_to_anchor=(0.5, 1.05), ncol=3,
            fancybox=True, shadow=True)
ml1 = MultipleLocator(10)
ml2 = MultipleLocator(3600)

And this is what I have so far: enter image description here
Notice the lengthy label for the blue line. I’d like that to be labeled as “Pitch” instead of the file name. In which line can I do that?

Asked By: N.K.



You need to gain access of the legend() object and use set_text() to change the text values, a simple example:

plt.plot(range(10), label='Some very long label')
plt.plot(range(1,11), label='Short label')
L.get_texts()[0].set_text('make it short')

enter image description here

In your case, you are changing the first item in the legend, I am quite sure the 0 index in L.get_texts()[0] applies to your problem too.

Answered By: CT Zhu

Another way:

Answered By: Kamil Sindi

The answer by ksindi works for setting the labels, but as some others commented, it can break the legend colours when used with seaborn (in my case a scatterplot: the dots and text didn’t line up properly anymore). To solve this, also pass the handles to ax.legend.

# the legend has often numbers like '0.450000007', the following snippet turns those in '0.45'
label_list = []
for t in ax.get_legend_handles_labels():
    # the first result will be all handles, i.e. the dots in the legend
    # the second result will be all legend text

new_list = []
for txt in label_list[1]:
    if txt[0] == '0':
        txt = str(txt)[:4]
label_list[1] = new_list

ax.legend(handles=label_list[0], labels=label_list[1])

(I would have posted this as a comment, but don’t have enough reputation yet)

Answered By: spatbord