Book Image

NumPy: Beginner's Guide

By : Ivan Idris
Book Image

NumPy: Beginner's Guide

By: Ivan Idris

Overview of this book

Table of Contents (21 chapters)
NumPy Beginner's Guide Third Edition
Credits
About the Author
About the Reviewers
www.PacktPub.com
Preface
NumPy Functions' References
Index

Time for action – plotting a polynomial and its derivatives


Let's plot a polynomial and its first and second derivative. We will make three subplots for the sake of clarity:

  1. Create a polynomial and its derivatives using the following code:

    func = np.poly1d(np.array([1, 2, 3, 4]).astype(float))
    x = np.linspace(-10, 10, 30)
    y = func(x)
    func1 = func.deriv(m=1)
    y1 = func1(x)
    func2 = func.deriv(m=2)
    y2 = func2(x)
  2. Create the first subplot of the polynomial with the subplot() function. The first parameter of this function is the number of rows, the second parameter is the number of columns, and the third parameter is an index number starting with 1. Alternatively, combine the three parameters into a single number, such as 311. The subplots will be organized in three rows and one column. Give the subplot the title Polynomial. Make a solid red line:

    plt.subplot(311)
    plt.plot(x, y, 'r-')
    plt.title("Polynomial")
  3. Create the third subplot of the first derivative with the subplot() function. Give the subplot...