numpy if A > Value, change B to constant

user_6456971 0
Viewed 92

Problem:

I've got a function with two arrays, but if the input data and result are within certain limits, the returned value is a constant:

def funcB(A, C1, C2):
    B = (0.381*A) + (0.05*(C1/C2)) - 0.15
    B[B > 1.0] = 1.0
    B[B < 0.5] = 0.5
    return B

In the above B is the result, if it is > 1.0 then it is 1, and < 0.5 is 0.5.

What I am struggling to do is also add a condition where if the input A is < 1.64 result B is also 0.5, and where if it is > 3.3 result B is 1.0

boolean numpy python
1 Answers

Reti43 - that worked!

Solution was as simple as:

def FuncB(A, C1, C2):
B = ((0.381*A) + (0.05*(C1/C2)) - 0.15).reshape(A.shape)
B[B > 1.0] = 1.0
B[B < 0.5] = 0.5
B[A < 1.64] = 0.5
B[A > 3.3] = 1.0
return B
user_64569710
Related