Creating a 5x5 array with random values and finding row maximum Indices using NumPy

Last update on December 21 2024 07:36:07 (UTC/GMT +8 hours)

Write a NumPy program to create a 5x5 array with random values and find the index of the maximum value in each row.

This NumPy program creates a 5x5 array filled with random values and then finds the index of the maximum value in each row. It iterates through each row to determine the index of the maximum value and stores the results in an array.

Sample Solution:

Python Code:

import numpy as np

# Create a 5x5 array with random values
array = np.random.random((5, 5))

# Find the index of the maximum value in each row
max_indices = np.argmax(array, axis=1)

# Print the array and the indices of the maximum values
print("Array:\n", array)
print("Indices of the maximum values in each row:\n", max_indices)

Output:

Array:
 [[0.03319994 0.87733658 0.15561569 0.09115729 0.50300971]
 [0.16636484 0.11402533 0.27361528 0.05784223 0.95653183]
 [0.9419271  0.03121247 0.19712034 0.92254315 0.34829598]
 [0.87937086 0.50750646 0.40539581 0.82764216 0.01438484]
 [0.19968155 0.57303488 0.94165787 0.44044397 0.87607812]]
Indices of the maximum values in each row:
 [1 4 0 0 2]

Explanation:

Import NumPy: Import the NumPy library to work with arrays.
Create a Random 5x5 Array: Generate a 5x5 array filled with random values using np.random.random.
Find Indices of Maximum Values: Use np.argmax with axis=1 to find the indices of the maximum values in each row.
Print Results: Print the generated array and the indices of the maximum values in each row.

Python-Numpy Code Editor: