w3resource

NumPy: Get the row numbers in given array where at least one item is larger than a specified value

NumPy: Array Object Exercise-151 with Solution

Write a NumPy program to get the row numbers in given array where at least one item is larger than a specified value.

Sample Solution:

Python Code:

import numpy as np
num = np.arange(36)
arr1 = np.reshape(num, [4, 9])
print("Original array:")
print(arr1)
result  = np.where(np.any(arr1>10, axis=1))
print("\nRow numbers where at least one item is larger than 10:")
print(result)

Sample Output:

Original array:
[[ 0  1  2  3  4  5  6  7  8]
 [ 9 10 11 12 13 14 15 16 17]
 [18 19 20 21 22 23 24 25 26]
 [27 28 29 30 31 32 33 34 35]]

Row numbers where at least one item is larger than 10:
(array([1, 2, 3]),)

Pictorial Presentation:

NumPy: Get the row numbers in given array where at least one item is larger than a specified value

Python Code Editor:

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

Previous: Write a NumPy program to swap columns in a given array.
Next: Write a NumPy program to calculate the sum of all columns of a 2D numpy array.

What is the difficulty level of this exercise?

Test your Python skills with w3resource's quiz



Python: Tips of the Day

Set comprehension:

>>> m = {x ** 2 for x in range(5)}
>>> m
{0, 1, 4, 9, 16}