Formatting and summing up numbers with arrays

Question

I am trying to calculate the T_Sum value so that for the values that are greater than the Vals values in Numbers it will just add up to the Vals values for that element. For example, the first element of Vals is 60 and all the values within Numbers are greater than 60 (11 in total), so the result will be 60 * 11. If the Vals value is 105 there are 5 elements that are greater than 105 so the result will be 525. How can I do this without a for loop?

Vals = np.arange(start=60, stop=105, step=5)
Numbers = np.array([123.6, 130, 150, 110.3748, 111.6992976,
 102.3165566, 97.81462811, 89.50038472, 96.48141473, 90.49956702, 65])

My attempt

T_Sum = np.ma.masked_array(np.repeat(Numbers[None,:],Vals.size,0),mask=[Numbers<Vals[:,None]]).sum(-1).data

Expected Output

[660, 650, 700, 750, 800, 850, 810, 760, 600, 525]

'there are 5 elements that are greater than 6' what do you mean by that? — Chris
– Chris, Commented Mar 3, 2021 at 23:47
Sorry about that i have updated it, i think it should be good now. — ege Selcuk
– ege Selcuk, Commented Mar 4, 2021 at 2:55

Reti43 · Accepted Answer · 2021-03-04 03:19:15Z

1

The end value of np.arange must be greater than 105, because it's not end inclusive.

Vals = np.arange(60, 106, 5)
T_Sum = (Numbers[:,None] > Vals).sum(axis=0) * Vals

answered Mar 4, 2021 at 3:19

Reti43

9,8373 gold badges30 silver badges48 bronze badges

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

Comments

Chris · Accepted Answer · 2021-03-04 00:16:48Z

-1

The arrange method is not inclusive of the stop value, so your Vals actually ends at 100, not 105 like you seem to think ti does.

In any case, this will do what you want, just have to play with the start/stop values to get the final result you are looking for.

import numpy as np
Vals= np.arange(start=60, stop=105, step=5)
from collections import Counter
Numbers = np.array([123.6,       130 ,       150,        110.3748,     111.6992976,
 102.3165566,   97.81462811 , 89.50038472 , 96.48141473 , 90.49956702, 65])


output = []

for v in Vals:
    c = 0
    for n in Numbers:
        if n>v:
            c+=1
    print(f'There are {c}, values greater than {v}')
    output.append(c*v)

print(output)

Output

There are 11, values greater than 60
There are 10, values greater than 65
There are 10, values greater than 70
There are 10, values greater than 75
There are 10, values greater than 80
There are 10, values greater than 85
There are 9, values greater than 90
There are 8, values greater than 95
There are 6, values greater than 100
[660, 650, 700, 750, 800, 850, 810, 760, 600]

edited Mar 4, 2021 at 0:16

answered Mar 3, 2021 at 23:52

Chris

16.3k3 gold badges26 silver badges41 bronze badges

1 Comment

ege Selcuk Over a year ago

I am trying to use a numpy function to do that, since for lops arent optimized well and are slow, but thanks anyways.

Collectives™ on Stack Overflow

Formatting and summing up numbers with arrays

2 Answers 2

Comments

1 Comment

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

1 Comment

Your Answer

Sign up or log in

Post as a guest

Linked

Related