Book Image

NumPy Essentials

By : Leo (Liang-Huan) Chin, Tanmay Dutta, Shane Holloway
Book Image

NumPy Essentials

By: Leo (Liang-Huan) Chin, Tanmay Dutta, Shane Holloway

Overview of this book

In today’s world of science and technology, it’s all about speed and flexibility. When it comes to scientific computing, NumPy tops the list. NumPy gives you both the speed and high productivity you need. This book will walk you through NumPy using clear, step-by-step examples and just the right amount of theory. We will guide you through wider applications of NumPy in scientific computing and will then focus on the fundamentals of NumPy, including array objects, functions, and matrices, each of them explained with practical examples. You will then learn about different NumPy modules while performing mathematical operations such as calculating the Fourier Transform; solving linear systems of equations, interpolation, extrapolation, regression, and curve fitting; and evaluating integrals and derivatives. We will also introduce you to using Cython with NumPy arrays and writing extension modules for NumPy code using the C API. This book will give you exposure to the vast NumPy library and help you build efficient, high-speed programs using a wide range of mathematical features.
Table of Contents (16 chapters)
NumPy Essentials
Credits
About the Authors
About the Reviewers
www.PacktPub.com
Preface

Fourier transform application


In the previous sections, you learned how to use numpy.fft for a one and multi-dimensional ndarray, and saw the implementation details underneath the hood. Now it's time for some applications. In this section, we are going to use the Fourier transform to do some image processing. We will analyze the spectrum, and then we will interpolate the image to enlarge it to twice the size. First, let's download the exercise image from the Packt Publishing website blog post:  https://www.packtpub.com/books/content/python-data-scientists. Save the image to your local directory as scientist.png.

This is a RGB image, which means that, when we convert it to an ndarray, it will be three-dimensional. To simplify the exercise, we use the image module in matplotlib to read in the image and convert it to grayscale:

In [52]: from matplotlib import image 
In [53]: img = image.imread('./scientist.png') 
In [54]: gray_img = np.dot(img[:,:,:3], [.21, .72, .07]) 
In [55...