Producing a simple plot

Without further ado, let's create our first plot.

Let's say that we want to produce a simple line plot of the sine function, sin(x). We want the function to be evaluated at all points on the axis where 0 < x < 10. We will use NumPy's linspace function to create a linear spacing on the x axis, from x values 0 to 10, and a total of 100 sampling points:

In [3]: import numpy as np
In [4]: x = np.linspace(0, 10, 100)

We can evaluate the sin function at all points, xusing NumPy's sin function, and visualize the result by calling the plot function of plt:

In [5]: plt.plot(x, np.sin(x))

Did you try it yourself? What happened? Did anything show up?

The thing is, depending on where you are running this script, you might not see anything. There are the following possibilities to consider:

  • Plotting from a .py script: If you are running matplotlib from a script, you need to simply call plt as follows:
         plt.show()

Call this at the end and your plot will show up!

  • Plotting from an IPython shell: This is actually one of the most convenient ways to run matplotlib interactively. To make the plots show up, you need to use what is called the %matplotlib magic command after starting IPython:
          In [1]: %matplotlib
Using matplotlib backend: Qt5Agg
In [2]: import matplotlib.pyplot as plt

Then, all of your plots will show up automatically without having to call plt.show() every time.

  • Plotting from a Jupyter Notebook: If you are viewing this code from a browser-based Jupyter Notebook, you need to use the same %matplotlib magic. However, you also have the option of embedding graphics directly in the notebook, with two possible outcomes:
    • %matplotlib notebook will lead to interactive plots embedded within the notebook.
    • %matplotlib inline will lead to static images of your plots embedded within the notebook.

In this book, we will generally opt for the inline option:

In [6]: %matplotlib inline

Now, let's try this again:

In [7]: plt.plot(x, np.sin(x))
Out[7]: [<matplotlib.lines.Line2D at 0x7fc960a80550>]

The preceding command gives the following output:

If you want to save the diagram for later, you have the option to do so directly from within IPython or Jupyter Notebook:

In [8]: savefig('figures/02.03-sine.png')

Just make sure that you use a supported file ending, such as .jpg, .png, .tif, .svg, .eps, or .pdf.

You can change the style of your plots by running plt.style.use(style_name) after importing matplotlib. All available styles are listed in plt.style.available. For example, try plt.style.use('fivethirtyeight'), plt.style.use('ggplot'), or plt.style.use('seaborn-dark'). For added fun, run plt.xkcd(), and then try and plot something else.
..................Content has been hidden....................

You can't read the all page of ebook, please click here login for view all page.
Reset
18.117.189.228