Advanced NumPy Exercises - Find the sum along the last axis of a 3x3x3 array

NumPy: Advanced Exercise-8 with Solution

Write a NumPy program to create a 3x3x3 array with random values and find the sum along the last axis.

Sample Solution:

Python Code:

import numpy as np
nums = np.random.rand(3, 3, 3)
print("Original array elements:")
print(nums)
sum_along_last_axis = np.sum(nums, axis=2)
print("\nSum along the last axis:")
print(sum_along_last_axis)

Sample Output:

Original array elements:
[[[0.66542849 0.8452917  0.14572805]
  [0.07553346 0.77602774 0.65256042]
  [0.64461145 0.80786691 0.29072733]]

 [[0.20943008 0.32606894 0.15482057]
  [0.13066528 0.36319837 0.83379825]
  [0.71210716 0.69634548 0.75775974]]

 [[0.14054546 0.61485173 0.42188979]
  [0.63975373 0.02786896 0.77550368]
  [0.81748415 0.15730395 0.86243833]]]

Sum along the last axis:
[[1.65644824 1.50412161 1.74320568]
 [0.69031959 1.32766189 2.16621238]
 [1.17728698 1.44312636 1.83722642]]

Explanation:

In the above exercise -

nums = np.random.rand(3, 3, 3): This creates a 3-dimensional array of shape (3, 3, 3) with random values between 0 and 1.

sum_along_last_axis = np.sum(nums, axis=2): This code calculates the sum along the last axis of the nums array, which has an index of 2. It creates a new 2-dimensional array of shape (3, 3) where each element is the sum of the corresponding values in the original array along the last axis.

Python-Numpy Code Editor:

Have another way to solve this solution? Contribute your code (and comments) through Disqus.

Previous: Normalize a 5x5 array row-wise with random values.
Next: Create a 5x5 array with random values and sort each row.