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

NumPy cbrt() Function



The NumPy cbrt() function is used to compute the cube root of all elements in an input array. It calculates ∛ for each element x in the array.

This function can be applied to scalars, lists, or NumPy arrays and will return an array of the same shape with the cube root of each input value.

Syntax

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

numpy.cbrt(x, /, out=None, where=True, casting='same_kind', order='K', dtype=None, subok=True[, signature, extobj])

Parameters

This function accepts the following parameters −

  • x: The input array or scalar. The function computes the cube root of each element of the array or scalar.
  • 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 result will be computed. Otherwise, the result 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 processed. The dtype of x 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 an array where each element is the cube root of the corresponding element in the input array x. If out is provided, it returns a reference to out.

Example: Basic Usage of cbrt() Function

In the following example, we use the cbrt() function to compute the cube root for each element in a 1-dimensional array −

import numpy as np

# Creating a 1-dimensional array
arr = np.array([1, 8, 27, 64])

# Applying cbrt to each element
result = np.cbrt(arr)
print(result)

The output obtained will be −

[1. 2. 3. 4.]

Example: cbrt() Function with Broadcasting

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

import numpy as np

# Creating a 2-dimensional array
arr = np.array([[1, 8, 27], [64, 125, 216]])

# Applying cbrt to each element
result = np.cbrt(arr)
print(result)

This will produce the following result −

[[1. 2. 3.]
 [4. 5. 6.]]

Example: cbrt() Function with Scalar

In this example, we apply the cbrt() function to a scalar value −

import numpy as np

# Scalar value
scalar = 27

# Applying cbrt to the scalar
result = np.cbrt(scalar)
print(result)

The output obtained is −

3.0000000000000004

Example: cbrt() Function with Negative Values

In this example, we apply the cbrt() function to an array with negative values −

import numpy as np

# Creating a 1-dimensional array with negative values
arr = np.array([-1, -8, -27])

# Applying cbrt to each element
result = np.cbrt(arr)
print(result)

The output obtained will be −

[-1. -2. -3.]
numpy_arithmetic_operations.htm
Advertisements