Das deutsche Python-Forum

Hallo zusammen

Ich habe ein numpy.array wie folgt:

>>> To
array([[ 1.5017,  0.    ],
       [ 0.    ,  0.4983]])

Ich möchte jetzt gerne die inverse Quadratwurzel nur von den Diagonalelementen nehmen (^(-1/2)).

Code: Alles auswählen

>>> 1/numpy.sqrt(To)
Warning: divide by zero encountered in divide
array([[  8.16034292e-001,               inf],
       [              inf,   1.41662387e+000]])

Tja, etwas schlaueres als numpy.ravel(To) - for loop mit if(diagonalelement) - numpy.reshape() fällt mir nicht ein.

Wäre toll, wenn mir jemand einen schlauen Hinweis geben könnte.

Code: Alles auswählen

>>> a = numpy.array([[1.5017, 0.], [0., 0.4983]])
>>> numpy.where(numpy.identity(len(a)), 1 / numpy.sqrt(a), a)
array([[ 0.81603429,  0.        ],
       [ 0.        ,  1.41662387]])

MfG
HWK

wunderbar, thanks.

Das deutsche Python-Forum

Numpy sqrt

Numpy sqrt