how to add legend to the plot?

[jainraje]

How to add a legend to the plots? For example, if using several moving averages it will be useful to show a legend to map moving averages to line plots.

Is clear how this is done using matplotlib but I did not see an example of how to do so using the mplfinance package.

TIA,

[DanielGoldfarb]

Rajeev,
Legends are not yet available in mplfinance. Hopefully soon (maybe a couple months or so).

In the meantime, if you want to play with matplotlib, you can try using the returnfig=True kwarg in mplfinance and attempt to display legends by having access to the Figure and Axes that mplfinance creates.

Alternatively if you would like to contribute code to mplfinance to make it easy for users to include legends, I would be happy to work together with you on getting that code correctly merged into mplfinance.

All the best. Thanks for you interest in mplfinance. --Daniel

[jainraje]

Hi Daniel,
I appreciate your comments. I'll give it a try. This package seems to be going in a good direction. Can we have a side bar on how I can contribute code to the package? You can email me directly at jainraje@yahoo.com.

All the best
--Rajeev

[arthurpdesimone]

Plot chart to figure and axis object

    fig, axes = mpf.plot(df, type='candle', mav=mav_tuple, returnfig=True)
    # Configure chart legend and title
    axes[0].legend(mav_titles)
    axes[0].set_title(ticker)
    # Save figure to file
    fig.savefig(path_to_figure)

This does the trick thanks to the help of mr. Daniel Goldfarb.

[DanielGoldfarb]

@arthurpdesimone
Thank you for posting a specific example to help others. Much appreciated. All the best. --Daniel

[YoungilKoh]

@arthurpdesimone Thank you for posting a specific example to help others. Much appreciated. All the best. --Daniel

Hi, I'm Youngil Koh

I think the sample code below would be helpful.

style = mpf.make_mpf_style(marketcolors=mpf.make_marketcolors(up="r", down="#0000CC",inherit=True),
                           gridcolor="gray", gridstyle="--", gridaxis="both")                    

added_plots = {"EMA112" : mpf.make_addplot(data['EMA112'].iloc[start:]),
               "EMA224" : mpf.make_addplot(data['EMA224'].iloc[start:]),
               "BBLow" : mpf.make_addplot(data['BBLow'].iloc[start:]),
               "Close" : mpf.make_addplot(data['Close'].iloc[start:])
}

fig, axes = mpf.plot(data.iloc[start:, 0:5], type="candle", style=style,
                     addplot=list(added_plots.values()),
                     volume=True,
                     returnfig=True)
                     
axes[0].legend([None]*(len(added_plots)+2))
handles = axes[0].get_legend().legendHandles
axes[0].legend(handles=handles[2:],labels=list(added_plots.keys()))

axes[0].set_ylabel("Price [KRW]")
axes[0].yaxis.set_major_formatter(StrMethodFormatter('{x:,.0f}'))
axes[2].set_ylabel("Volume [ea]")
axes[2].yaxis.set_major_formatter(StrMethodFormatter('{x:,.0f}'))

[DanielGoldfarb]

@YoungilKoh
Thank you so much! That is an excellent example! Much appreciated.

[BlackSharkJ]

@arthurpdesimone Thank you for posting a specific example to help others. Much appreciated. All the best. --Daniel

Hi, I'm Youngil Koh

I think the sample code below would be helpful.

style = mpf.make_mpf_style(marketcolors=mpf.make_marketcolors(up="r", down="#0000CC",inherit=True),
                           gridcolor="gray", gridstyle="--", gridaxis="both")                    

added_plots = {"EMA112" : mpf.make_addplot(data['EMA112'].iloc[start:]),
               "EMA224" : mpf.make_addplot(data['EMA224'].iloc[start:]),
               "BBLow" : mpf.make_addplot(data['BBLow'].iloc[start:]),
               "Close" : mpf.make_addplot(data['Close'].iloc[start:])
}

fig, axes = mpf.plot(data.iloc[start:, 0:5], type="candle", style=style,
                     addplot=list(added_plots.values()),
                     volume=True,
                     returnfig=True)
                     
axes[0].legend([None]*(len(added_plots)+2))
handles = axes[0].get_legend().legendHandles
axes[0].legend(handles=handles[2:],labels=list(added_plots.keys()))

axes[0].set_ylabel("Price [KRW]")
axes[0].yaxis.set_major_formatter(StrMethodFormatter('{x:,.0f}'))
axes[2].set_ylabel("Volume [ea]")
axes[2].yaxis.set_major_formatter(StrMethodFormatter('{x:,.0f}'))

Thank you for your help. You're amazing

[quebeh]

I found an issue while following @YoungilKoh 's answer (#181 (comment))

It worked, but only one of the plots can display (see attachment).

I'll need help on why only one of them is showing in the legends.

I debugged it, and i think i know the issue, the legendHandles only gives one Line2D object when printed, this might be a hint that can help us find the problem