Hinzufügen Submatrix in Numpy ohne Schleife

Question

Lasst uns sagen, dass ich diese bekam a = np.arange(9).reshape((3,3)) ich eine numpy Reihe von [9,12,15] erhalten möchten, die ein Ergebnis von

[0+3+6, 1+4+7, 2+5+8] 

Answer 1

Sie numpy.array.sum() Funktion, indem man die axis=0 ist verwenden können:

>>> a.sum(axis=0) 
array([ 9, 12, 15]) 

Answer 2

Verwenden Sie die numpy.sum-Funktion und geben Sie die Achse an, über die Sie summieren möchten. Dies ist 0 in Ihrem Fall:

import numpy as np 

a = np.arange(9).reshape((3,3)) 
a_sum = np.sum(a, axis=0) 

print a_sum 

Dies gibt Ihnen:

[ 9 12 15] 

Die Antwort von Kasramvd verwendet den objektorientierten Ansatz, der einige Leute bevorzugen:

a_sum = a.sum(axis=0)