Diffusion d'un tableau avec numpy

Question

Comment écrire la boucle suivante en utilisant la boucle implicite de Python?

def kl(myA, myB, a, b): 
    lots of stuff that assumes all inputs are scalars 

x, y = meshgrid(inclusive_arange(0.0, xsize, 0.10),\ 
       inclusive_arange(0.0, ysize, 0.10)) 

for j in range(x.shape[0]): 
    for i in range(x.shape[1]): 
     z[j, i] = kl(x[j, i], y[j, i]) 

Je veux faire quelque chose comme

z = kl(x, y) 

mais qui donne:

TypeError: only length-1 arrays can be converted to Python scalars 

Answer 1

La capacité que vous vous posez au sujet existe que dans Numpy, et il est appelé array broadcasting, non implicite en boucle. Une fonction qui diffuse une opération scalaire sur un tableau s'appelle une fonction universelle, ou ufunc. De nombreuses fonctions de base de Numpy sont de ce type.

Vous pouvez utiliser numpy.frompyfunc pour convertir votre fonction existante kl en un ufunc.

kl_ufunc = numpy.frompyfunc(kl, 4, 1) 
... 
z = kl_ufunc(x + 1.0, y + 1.0, myA, myB) 

Bien sûr, si vous voulez, vous pourriez appeler le ufunc kl au lieu de kl_ufunc, mais la définition originale de kl serait perdu. Cela pourrait convenir à vos objectifs.

Answer 2

Il y a une série de vidéos ici que vous trouverez peut-être utile:

http://showmedo.com/videotutorials/video?name=10370070&fromSeriesID=1037

Notez qu'il fait partie d'une série de tutoriel qui traite d'un large éventail de sujets numpy.

Juste pour info.