Numpy Gradient: Returning the Gradient of N-dimensional Array

Numpy Gradient

Hate to break this, but Mathematics is tiresome! It is a lot more intense than one can think of, especially when one enters the territory of differentiation and integration. To make things worse, the differentiation is flexible enough to be done partially.

Good Lord above heavens! But hey, we have got Python to carry out some of the stuff for us. Phew!

This article explains on the deployment of the gradient( ) function within the numpy library of Python for usage against the arrays of N-dimensions.

Also read: NumPy nanmax – Maximum of an array along an axis ignoring any NaNs


Syntax of numpy.gradient()

Let us first have a look at the syntax of the gradient( ) function before getting to know the hows and whats of using it.

numpy.gradient( array, varags, axis=None, edge_order=1)

where,

  • array – a collection of scalar entities of N-dimensions
  • varags – an optional provision to include the variable arguments which dictate the spacing for each dimension within the input array
  • axis – an optional provision that dictates the direction for calculating the gradient & is set to none by default
  • edge_order – an optional provision that deals with the boundaries at which the gradient is to be calculated. It can be set as ‘1’ or ‘2’, with the former being the default setting

Python calculates the gradient by finding the difference,

  • between the adjacent numbers at boundaries & dividing by default spacing ‘1’
  • between the adjacent numbers in the interior & dividing by default spacing ‘2’

Calculating Gradient with Uniform Spacing:

After importing the numpy library let us start by constructing an array & then finding its gradient with uniform spacing, say ‘3’ as given below.

import numpy as np
ar = np.array([1.2, 3.4, 5.6])
np.gradient(ar,3)
Gradient Calculated With Uniform Spacing
Gradient Calculated With Uniform Spacing

The same can also be done with the default setting of the gradient( ) function without explicitly mentioning the required spacing. In this case, the spacing is set to the default values stated earlier in the syntax section of this article. Also, it is to be noted that one can also specify the data type as ‘float’ within the array in case one ought to find the gradient for a bunch of whole numbers.

ar = np.array([1, 3, 5, 9], dtype = float)
np.gradient(ar)
Calculating Gradient With Default Values
Calculating Gradient With Default Values

Calculating Gradient with Non-Uniform Spacing

One can also use exclusive spacing by assigning an array with the required spacing for each element of the input array within the gradient( ) function. But, one ought to bear in mind that this array should be of the same dimension as that of the input array.

ar = np.array([1.2, 3.4, 5.6], dtype = float)
sp = np.array([7.8, 9.0, 0.1], dtype = float)
np.gradient(ar,sp)
Calculating Gradient With Non Uniform Spacing
Calculating Gradient With Non-Uniform Spacing

Calculating Gradient for N-Dimensional Array:

When a multi-dimensional array arrives into the picture, the gradient( ) function shall return two different results irrespective of whether one provides uniform or non-uniform spacing. This leaves us wondering ‘Now, why would that happen?!’.

This outcome can be attributed to the fact that one result corresponds to the gradient calculation with respect to the rows and the other corresponds to the gradient calculation with respect to the columns of the input array.

Let us try calculating the gradient of a two-dimensional array with a uniform spacing of ‘2’, as shown below.

ar =([[1.2, 3.4, 5.6], [7.8, 9.0, 0.1]])
np.gradient(ar,2)
Gradient Of 2 Dimensional Array With Uniform Spacing
Gradient Of 2 Dimensional Array With Uniform Spacing

The first result in the above image is the gradient calculated with respect to the columns and the one that follows is the gradient calculated with respect to the rows.


Conclusion:

Now that we have reached the end of this article, hope it has elaborated on how to find the gradient of an N-dimensional array in Python. Here’s another article that explains how to return the reciprocal of each element using numpy in Python. There are numerous other enjoyable & equally informative articles in AskPython that might be of great help to those who are looking to level up in Python. Whilst you enjoy those, hasta luego!