Location via proxy:   [ UP ]  
[Report a bug]   [Manage cookies]                

NumPy add() Function



The NumPy add() function is used to perform element-wise addition of two arrays. This function takes two input arrays of the same shape and returns a new array with each element being the sum of the corresponding elements in the input arrays.

If the input arrays have different shapes, broadcasting rules apply to make their shapes compatible. This means that the function can handle arrays of different shapes under certain conditions.

Syntax

Following is the syntax of the NumPy add() function −

numpy.add(x1, x2, /, out=None, *, where=True, casting='same_kind', order='K', dtype=None, subok=True[, signature, extobj])

Parameters

This function accepts the following parameters −

  • x1: The first input array. Both arrays should have the same shape, or they must be broadcastable to a common shape.
  • x2: The second input array. Like x1, it should have the same shape as x1, or be broadcastable to a common shape.
  • out (optional): A location into which the result is stored. If provided, it must have a shape that the inputs broadcast to. If not provided or None, a freshly-allocated array is returned.
  • where(optional): This condition is broadcast over the input. At locations where the condition is True, the out array will be set to the ufunc result. Otherwise, it will retain its original value.
  • casting(optional): Controls what kind of data casting may occur. Defaults to 'same_kind'.
  • order(optional): Controls the memory layout order of the result. 'C' means C-order, 'F' means Fortran-order, 'A' means 'F' if inputs are all F, 'C' otherwise, 'K' means match the layout of the inputs as closely as possible.
  • dtype(optional): The type of the returned array and of the accumulator in which the elements are summed. The dtype of x1 and x2 is used by default unless dtype is specified.
  • subok(optional): If True, then sub-classes will be passed-through, otherwise the returned array will be forced to be a base-class array.

Return Value

This function returns the element-wise sum of the input arrays x1 and x2. If out is provided, it returns a reference to out.

Example: Basic Usage of add() Function

In the following example, we create two 1-dimensional arrays and use the add() function to perform element-wise addition −

import numpy as np

# Creating two 1-dimensional arrays
arr1 = np.array([1, 2, 3, 4])
arr2 = np.array([5, 6, 7, 8])

# Performing element-wise addition
result = np.add(arr1, arr2)
print(result)

Following is the output obtained −

[ 6  8 10 12]

Example: Adding Arrays with Broadcasting

In this example, we demonstrate the use of broadcasting with the add() function. We create a 2-dimensional array and add a 1-dimensional array to it −

import numpy as np

# Creating a 2-dimensional array
arr1 = np.array([[1, 2, 3], [4, 5, 6]])

# Creating a 1-dimensional array
arr2 = np.array([10, 20, 30])

# Performing element-wise addition with broadcasting
result = np.add(arr1, arr2)
print(result)

This will produce the following result −

[[11 22 33]
 [14 25 36]]

Example: Using the "out" Parameter

In this example, we use the out parameter to store the result of the addition in a specified array −

import numpy as np

# Creating two 1-dimensional arrays
arr1 = np.array([1, 2, 3, 4])
arr2 = np.array([5, 6, 7, 8])

# Creating an output array
out_arr = np.empty(4)

# Performing element-wise addition and storing the result in out_arr
np.add(arr1, arr2, out=out_arr)
print(out_arr)

Following is the output of the above code −

[ 6.  8. 10. 12.]

Example: Using the "where" Parameter

In this example, we use the where parameter to conditionally perform the addition operation −

import numpy as np

# Creating two 1-dimensional arrays
arr1 = np.array([1, 2, 3, 4])
arr2 = np.array([5, 6, 7, 8])

# Creating a condition array
condition = np.array([True, False, True, False])

# Performing element-wise addition where the condition is True
result = np.add(arr1, arr2, where=condition)
print(result)

The output obtained is as shown below −

[ 6  0 10 16]
numpy_arithmetic_operations.htm
Advertisements