Adding input variables to plot title/legend in Python

Question

I would like to display the current value of a parameter used to plot a certain function in the plot title/legend/annotated text. As a simple example, let's take a straight line:

import numpy
import matplotlib.pyplot as plt

def line(m,c):
   x = numpy.linspace(0,1)
   y = m*x+c
   plt.plot(x,y)
   plt.text(0.1, 2.8, "The gradient is" *the current m-value should go here*)
   plt.show()

print line(1.0, 2.0)

In this case, I would like my text to say "The gradient is 1.0", but I'm not sure what the syntax is. Moreover, how would I include the second (and more) parameter(s) below, so that it reads:

"The gradient is 1.0

The intercept is 2.0."

N Chauhan · Accepted Answer · 2018-08-12 19:57:21Z

10

Use string formatting with the .format() method:

plt.text(0.1, 2.8, "The gradient is {}, the intercept is {}".format(m, c))

Where m and c are the variables you want to substitute in.

You can directly write the variables like this in Python 3.6+ if you prefix the string with an f whcih denotes a formatted string literal:

f"the gradient is {m}, the intercept is {c}"

answered Aug 12, 2018 at 19:57

N Chauhan

3,5152 gold badges9 silver badges23 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

blue note · Accepted Answer · 2018-08-12 19:56:51Z

3

In python 3.6+ you can do it by prefixing the string with f, and putting the variable in curly brackets. For earlier python version there have been various ways of doing it, look up string formatting

message = f"The slope is {m}"
plt.text(message)

(by the way, gradient is usually called slope when referring to single variable linear equation)

answered Aug 12, 2018 at 19:56

blue note

29.5k10 gold badges83 silver badges110 bronze badges

Comments

Eric Aya · Accepted Answer · 2022-11-18 02:59:15Z

0

The other answers didn't work for my code, but adaption of it did. Shown below:

Showing y = m*x + c printed on the plot in log format.

a1 = coefs[0] # variable 1
a2 = coefs[1] # variable 2

message = f"log(L/Lo) = {a1} * log(M/Mo) + {a2}"

# Define axes
left = 0.01
width = 0.9
bottom  = 0.01
height = 0.9
right = left + width
top = bottom + height
ax = plt.gca()

# Transform axes
ax.set_transform(ax.transAxes)

# Define text
ax.text(0.5 * (left + right), 0.5 * (bottom + top), message,
        horizontalalignment='center',
        verticalalignment='center',
        size= 10,
        color='r',
        transform=ax.transAxes)

plt.show()

Using code from @ https://pythonguides.com/add-text-to-plot-matplotlib/

edited Nov 18, 2022 at 2:59

Eric Aya

70.2k36 gold badges190 silver badges266 bronze badges

answered Nov 18, 2022 at 0:13

Brendan Darrer

5095 silver badges13 bronze badges

Collectives™ on Stack Overflow

Adding input variables to plot title/legend in Python

3 Answers 3

Comments

Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

Comments

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related