Numpy Math functions

In this article, we will learn about basic NumPy math functions on array elements. We will learn with an example of how we can do addition, subtraction, multiplications, etc on NumPy arrays.

1. Addition

In this example, we will learn addition operation on NumPy array.We can Add a number to each array element as in example below:

In the output you can see each element got the addition of 1.

import numpy as np

arr = np.array([2,4,6,8])

# here adding 1 to each element
array_one = np.add(arr,1) 
print("After adding 1 to each element",array_one)

Output :

After adding 1 to each element [3 5 7 9]

2. Subtraction

We can Subtract a number from each array element

import numpy as np

arr = np.array([2,4,6,8])

array_two = np.subtract(arr,2) 

# here subtract 2 from each array element

print("After subtracting 2 from each element",array_two)

Output :

After subtracting 2 from each element [0 2 4 6]

3. Multiply of Numpy array element

We can Multiply a number to each array element


import numpy as np

arr = np.array([2,4,6,8])
array_three = np.multiply(arr,3) 

# here Multiply 3 to each array element

print("After multiply 3 to each element",array_three)

Output :

After multiply 3 to each element [ 6 12 18 24]

4. Power

We can do power to each element by using power function()

import numpy as np

arr = np.array([2,4,6,8])
array_four = np.power(arr,5) 

# here each array element to the 5th power

print("Doing power to 5 to each element",array_four)

Output :

Doing power to 5 to each element [ 32 1024 7776 32768]

5. Division

We can Divide each array element by a number. If we try division by 0 then it returns each element as inf

import numpy as np

arr = np.array([2,4,6,8])

# here Divide each array element by 4

array_five = np.divide(arr,4) 

print("After dividing by 4 to each element",array_five)

Output :

After dividing by 4 to each element [0.5 1. 1.5 2. ]

Dividing a NumPy array elements by zero

When you try this division then you may get a warning from the interpreter.

import numpy as np

arr = np.array([2,4,6,8])
array_six = np.divide(arr,0)
print("After dividing by 0 to each element",array_six)


Output :

RuntimeWarning: divide by zero encountered in true_divide
After dividing by 0 to each element [inf inf inf inf]