How to find maximum negative and minimum positive number in a numpy array?

Question

I am given an array containing both positive and negative numbers.

import numpy as np
arr = np.array([-10.2, -5.3, -2.1, 0, 1.2, 3.4])

I would like to find the index corresponding to the maximum negative number and to a minimum positive number. In the above, my expected outcome is 2 and 4. Is there any numpy trick to achieve this? I have found a solution in this link, but I would like to know how this could be done through numpy functions: Finding index of largest negative and smallest positive element in array

Is it guaranteed that the input is sorted?

John Zwinck
– John Zwinck

2019-04-20 01:46:33 +00:00
Commented Apr 20, 2019 at 1:46 — John Zwinck
– John Zwinck, Commented Apr 20, 2019 at 1:46
No, it is not guaranteed.

matttree
– matttree

2019-04-20 02:36:33 +00:00
Commented Apr 20, 2019 at 2:36 — matttree
– matttree, Commented Apr 20, 2019 at 2:36

akuiper · Accepted Answer · 2019-04-20 01:53:07Z

21

Replace non negative values with -inf, then use argmax to find the largest negative:

np.where(arr < 0, arr, -np.inf).argmax()
# 2

Similarly replace non positive values with inf then use argmin to find the smallest positive:

np.where(arr > 0, arr, np.inf).argmin()
# 4

answered Apr 20, 2019 at 1:53

akuiper

216k33 gold badges362 silver badges379 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

Warren Weckesser Over a year ago

Or replace with nan and use nanargmax and nanargmin. But that's not much different than using inf.

Lost Over a year ago

@Psidom Is using Numpy functions as in your answer faster or doing it with lists as mentioned in the answer tagged in the question faster?

kmario23 · Accepted Answer · 2019-04-20 02:30:12Z

1

If your array is always sorted (as in the provided example), then:

# input
Out[66]: array([-10.2,  -5.3,  -2.1,   0. ,   1.2,   3.4])

# obtain a `signed` boolean mask 
In [75]: sign_mask = np.sign(arr)

# compute derivative and obtain index for max_negative element
In [76]: max_neg_idx = np.where(np.diff(sign_mask, append=1) == 1)[0][0]

# add +2 to that to get index for min_positive element
In [77]: min_pos_idx = max_neg_idx + 2

answered Apr 20, 2019 at 2:30

kmario23

62.1k17 gold badges174 silver badges159 bronze badges

Comments

Raghul Raj · Accepted Answer · 2020-04-12 13:39:13Z

0

I was facing a similar type of problem and this was my approach:

arr=np.array([-10.2, -5.3, -2.1, 0, 1.2, 3.4])

#Filtering for all the negative values and getting the min of them
print('Threshold for Min val',np.where(arr==[max(arr[arr<0])]))

#Filtering for all the positive values and getting the max of them
print('Threshold for Max val',np.where(val==[min(arr[arr>0])]))

answered Apr 12, 2020 at 13:39

Raghul Raj

1,44811 silver badges24 bronze badges

Collectives™ on Stack Overflow

How to find maximum negative and minimum positive number in a numpy array?

3 Answers 3

2 Comments

Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

2 Comments

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related