numba \ jit не позволяет использовать np argsort в режиме nopython - PullRequest
Новая
       9

numba \ jit не позволяет использовать np argsort в режиме nopython

0 голосов
/ 29 июня 2018

Получение этого сообщения об ошибке: 

Failed at nopython (nopython frontend)

[1m[1m[1mInvalid usage of Function(<function argsort at 0x0000000002A67840>) with parameters (array(float64, 2d, C), axis=int64)

 * parameterized

In definition 0:


При использовании этого кода 

def rankbids(bids, shifts, groupPeriod, period):
    rowsSize = bids.shape[0];

    finaltable = np.zeros((rowsSize, groupPeriod), dtype=np.float64)
    for i in range(0, period):
        #for 0 to 99
        #CONSTANT 4 UPDATE WHEN NEEDED


        for worker in range(rowsSize):
            shiftNum = int(shifts[worker,i]);

            finaltable[worker, (shiftNum+i*10)] = bids[worker,i];

            if shiftNum == 1:
                finaltable[worker, (shiftNum+i*10)] = bids[worker,i];
                finaltable[worker, (shiftNum+1+i*10)] = bids[worker,i];
                finaltable[worker, (shiftNum+2+i*10)] = bids[worker,i];
            elif shiftNum == 2:
                finaltable[worker, (shiftNum+2+i*10)] = bids[worker,i];
                finaltable[worker, (shiftNum+3+i*10)] = bids[worker,i];
                finaltable[worker, (shiftNum+4+i*10)] = bids[worker,i];
            elif shiftNum == 3:
                finaltable[worker, (shiftNum+4+i*10)] = bids[worker,i];
                finaltable[worker, (shiftNum+5+i*10)] = bids[worker,i];
                finaltable[worker, (shiftNum+6+i*10)] = bids[worker,i];


    indexTable = np.argsort(finaltable, axis=0)
    print(finaltable);




    return finaltable;

rank_bids = numba.jit('float64[:,:](float64[:,:], float64[:,:], int64, float64)', nopython=True)(rankbids);

Кажется, что numpy не разрешает argsort с массивами Nd в функциях numba? Мой вопрос: смог ли кто-нибудь использовать его в функции jit и, возможно, показать мне, что я мог бы сделать, чтобы использовать его!

1 Ответ

0 голосов
/ 29 июня 2018

np.argsort работает в numba, но не в ключевом слове axis. Вы можете написать свой код indexTable = np.argsort(finaltable, axis=0) так: 

indexTable = np.empty_like(finaltable)
for j in range(indexTable.shape[1]):
    indexTable[:, j] = np.argsort(finaltable[:, j])
...