Summenarray mit Zustand in einem anderen Array mit numpy

Question

ich folgende Python-Code haben ein Array mit Zustand in einem anderen Array

sum=0 
for i in range(grp_num): 
    if lower_bounds[i] > 0: 
     sum = sum + histo1[i] 

I numpy Äquivalent glauben zusammenzufassen wäre np.where(lower_bounds>0, histo1,0).sum() Aber die numpy Methode summiert sich alles in histo1 (ignoriert die Anforderung, dass lower_bounds> 0). Warum? Oder gibt es einen anderen Weg, dies zu tun? Vielen Dank.

Answer 1

Ok, dann ist dies zwar Vermutungen, aber die einzige Erklärung, die ich von für Ihren np.where(lower_bounds>0, histo1,0).sum() denken kann die volle Summe zurückkehr

• Sie auf Python2 sind
• lower_bounds ist eine Liste, kein Array

auf Python2:

[1, 2] > 0 
True 

Bedeutung dass Ihre numpy Zeile das erste Argument ausstrahlt und immer aus histo1, niemals aus 0 auswählt. Beachten Sie, dass die alternative Formulierung, die in den Kommentaren histo1[lower_bounds>0].sum() vorgeschlagen wurde, in dieser Situation auch nicht funktioniert (es wird histo1[1] zurückgegeben).

Die Lösung. Umwandeln von lower_bounds in ein Array explizit

np.where(np.array(lower_bounds)>0, histo1, 0) 

Btw. auf Python3 würden Sie eine Ausnahme

[1, 2] > 0 
Traceback (most recent call last): 
    File "<stdin>", line 1, in <module> 
TypeError: '>' not supported between instances of 'list' and 'int'