Numba autojit errore sul confronto array NumPy

Question

Quando metto a confronto due array NumPy dentro la mia funzione ottengo un errore che dice solo di lunghezza-1 array possono essere convertiti in scalari Python:

from numpy.random import rand 
from numba import autojit 

@autojit 
def myFun(): 
    a = rand(10,1) 
    b = rand(10,1) 
    idx = a > b 
    return idx 

myFun() 

L'errore:

--------------------------------------------------------------------------- 
TypeError         Traceback (most recent call last) 
<ipython-input-7-f7b68c0872a3> in <module>() 
----> 1 myFun() 

/Users/Guest/Library/Enthought/Canopy_64bit/User/lib/python2.7/site-packages/numba/numbawrapper.so in numba.numbawrapper._NumbaSpecializingWrapper.__call__ (numba/numbawrapper.c:3764)() 

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

Answer 1

Questo potrebbe essere secondario al tuo problema, ma il modo in cui hai mostrato autojit non ti farà aumentare la velocità. Con numba è necessario mostrare esplicitamente le for loop in questo modo:

from numpy.random import rand 
from numba import autojit 
@autojit 
def myFun(): 
    a = rand(10,1) 
    b = rand(10,1) 
    idx = np.zeros((10,1),dtype=bool) 
    for x in range(10): 
     idx[x,0] = a[x,0] > b[x,0] 
    return idx 

myFun() 

Questo funziona bene.