NumPy: Get the powers of an array values element-wise

Last update on March 29 2025 13:04:47 (UTC/GMT +8 hours)

5. Element-wise Exponentiation

Write a NumPy program to get the powers of an array values element-wise.

Note: First array elements raised to powers from second array.

Sample Solution:

Python Code:

# Importing the NumPy library
import numpy as np

# Creating an array from 0 to 6
x = np.arange(7)

# Displaying the original array
print("Original array:")
print(x)

# Raising the elements of the array to the power of 3 and displaying the result
print("First array elements raised to powers from second array, element-wise:")
print(np.power(x, 3)) 

Sample Output:

Original array                                                         
[0 1 2 3 4 5 6]                                                        
First array elements raised to powers from second array, element-wise: 
[  0   1   8  27  64 125 216]

Explanation:

In the above exercise –

x = np.arange(7): This line creates a NumPy array that includes numbers from 0 to 6.

print(np.power(x, 3)): This line takes the above array and raises each element to the power of 3. The output of this code will be a new NumPy array where each element is the cube of the corresponding element in the original x array. That is, it would output the array [0, 1, 8, 27, 64, 125, 216].

Pictorial Presentation:

For more Practice: Solve these Related Problems:

• Implement a function that raises elements of one array to the powers specified by a second array using np.power.
• Test element-wise exponentiation on arrays with zero, negative, and fractional bases with various exponents.
• Use broadcasting to compute powers when the exponent array has a different shape than the base array.
• Compare the output of np.power with a loop-based computation to verify correctness.

Python-Numpy Code Editor: