Matplotlib legend in subplot

In this tutorial, we will learn how Matplotlib can be used to include legends in subplots. The legend can be added after creating the plot using the legend() function.

Syntax:

The syntax for legend in subplot is:

<br />
axes[position].legend(loc = ”)<br />

where, loc is used for location.

Approaches:

Following are the approaches we will use for displaying legend in the subplot in Matplotlib:

First, we will use the subplots() function for creating subplots in a bigger plot.
Then, we will use the legend() function for adding a label to the curves.
At last, we will show the plots by using the show()

Example 1:

In this example, we will plot the scatter graph using subplot of log10 and exponential:

<br />
# First, we will import the required modules<br />
from matplotlib import pyplot as PPlt<br />
import numpy as nmp</p>
<p># here we will assign the value to x axis<br />x_axis1 = nmp.arange(2, 22, 1.5)</p>
<p># Now, we will get the value of log10<br />y_axis_log10_1 = nmp.log10(x_axis)</p>
<p># then, we will get the value of exponential<br />y_axix_exp1 = nmp.exp(x_axis)</p>
<p># Here, we will create subplots by using subplot() function<br />fig, axes = PPlt.subplots(2)</p>
<p># now, we will depicte the visualization<br />axes[0].plot(x_axis1, y_axis_log10_1, color = ‘Red’, label = “log10”)<br />axes[1].plot(x_axis1, y_axix_exp1, color = ‘Pink’, label = “exponential”)</p>
<p># Here, we will select the position at which legend to be added<br />axes[0].legend(loc = ‘best’)<br />axes[1].legend(loc = ‘best’)</p>
<p># At last, we will display the plot</p>
<p>PPlt.show()<br />

Output

Matplotlib legend in subplot

Example 2:

In this example, we will plot the scatter graph using subplot of sine and cos:

<br />
# First, we will import the required modules<br />
from matplotlib import pyplot as PPlt<br />
import numpy as nmp</p>
<p># here we will assign the value to x axis<br />x_axis1 = nmp.arange(-2, 3, 0.5)</p>
<p># Now, we will get the value of sine<br />y_axis_sine1 = nmp.sin(x_axis)</p>
<p># Now, we will get the value of cos<br />y_axix_cose1 = nmp.cos(x_axis)</p>
<p># Here, we will create subplots by using subplot() function<br />fig, axes = PPlt.subplots(2)</p>
<p># now, we will depicte the visualization<br />axes[0].scatter(x_axis1, y_axis_sine1, color = ‘Red’, marker = ‘*’, label = “sine”)<br />axes[1].scatter(x_axis1, y_axix_cose1, color = ‘Black’, marker = ‘*’, label = “cos”)</p>
<p># Here, we will select the position at which legend to be added<br />axes[0].legend(loc = ‘best’)<br />axes[1].legend(loc = ‘best’)</p>
<p># At last, we will display the plot<br />PPlt.show()<br />

Output

Matplotlib legend in subplot

Example 3:

In this example, we will plot the scatter graph using subplot of (y = x^2) and (y = x^3):

<br />
# First, we will import the required modules<br />
from matplotlib import pyplot as PPlt</p>
<p># here we will assign value to x axis<br />x_axis = list(range(-15, 15))</p>
<p># here, we will get the value of x * x<br />y_axis_1 = [x * x for x in x_axis]</p>
<p># here, we will get the value of x * x * x<br />y_axix_2 = [x * x * x for x in x_axis]</p>
<p># Now, we will create subplots by using subplot() function<br />fig, axes = PPlt.subplots(2)</p>
<p># Now, we will depicte the visualization<br />axes[0].scatter(x_axis, y_axis_1, color = ‘Blue’, marker = ‘*’, label = “y = x ^ 2”)<br />axes[1].scatter(x_axis, y_axix_2, color = ‘Brown’, marker = ‘*’, label = “y = x ^ 3”)</p>
<p># Here, we will select the position at which legend to be added<br />axes[0].legend(loc = ‘upper center’)<br />axes[1].legend(loc = ‘upper left’)</p>
<p># At last, we will display the plot<br />PPlt.show()<br />

Output

Matplotlib legend in subplot

Conclusion

In this tutorial, we have discussed how to use legend in the subplot of matplotlib graphs using different methods.

原创文章，作者：ItWorker，如若转载，请注明出处：https://blog.ytso.com/263297.html

Matplotlib legend in subplot

Matplotlib legend in subplot

Syntax:

Approaches:

Example 1:

Example 2:

Example 3:

Conclusion

相关推荐

发表回复