Task: Find Indices of Maximum Elements Along an Axis in NumPy Array

Task: Find Indices of Maximum Elements Along an Axis in NumPy Array: a task in Terminal-Lego-15k (Harbor dataset). You need to write a Python script that finds the indices of maximum elements in a 2D NumPy array along a specified axis.