Instagram
youtube
Facebook
Twitter

Compute Sum of Each Slice in 3D NumPy Array

A Compute Sum of Each Slice in 3D NumPy Array?
Code Explanation:
Library Import:
numpy is imported as np to perform array operations.
Create 3D Array: np.random.rand(2, 3, 4) generates a 3D NumPy array with shape (2, 3, 4) filled with random float     values.
Compute Sum: np.sum(arr, axis=(1, 2)) computes the sum over axes 1 and 2, which means summing all rows and         columns in each 2D slice.
Slice-wise Sum: The result is a 1D array containing the sum of elements in each 2D slice (block) of the 3D array.
Store Result: The result is stored in sum_each_slice.
Display Output: print(sum_each_slice) prints the sum of each 2D slice separately.

 

Program:

import numpy as np

arr = np.random.rand(2, 3, 4)

sum_each_slice = np.sum(arr, axis=(1, 2))

print(sum_each_slice)