get sum of numpy array when another array equals 1

Question

im trying to create a custom scoring function but I am having trouble with the last line.

def scorer(y_true, y_pred):
    y_pred1 = shift(y_pred,1)
    y_pred2 = shift(y_pred,2)
    target = y_pred - y_true
    action = np.where(target > 0, 1, 0)
    diff = y_pred2 - y_pred1
    score = (action == 1).sum(diff)

My output is supposed to be the sum of the diff rows when the action is equal to 1 for that row.

Diff	Action
1.28	1
2.56	0
.64	1
.32	0
5.12	0

For example, in this case it would be 1.28+.64=1.92

needs a stackoverflow.com/help/minimal-reproducible-example in particular, what is the output you see currently and what is the output you expect? — Lucas Roberts
– Lucas Roberts, Commented Apr 10, 2022 at 20:54

Lucas Roberts · Accepted Answer · 2022-04-10 21:02:03Z

1

You can put the diff directly into the np.where:

action = np.where(target > 0, y_pred2 - y_pred1, 0)

then sum action, via

sum(action)

answered Apr 10, 2022 at 21:02

Lucas Roberts

1,3531 gold badge15 silver badges18 bronze badges

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

Collectives™ on Stack Overflow

get sum of numpy array when another array equals 1

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related