def test(arr, k):
best_indexes = utils.argsort_best(arr, k, reverse=True)
print 'Array = %s' % arr
print 'Best indexes = %s' % best_indexes
print 'Best elements = %s' % arr[best_indexes]
print 'Number of indexes = %s' % len(best_indexes)
print 'Best element = %s' % np.max(arr)
print 'Took %.2f sec.' % utils.argsort_best.time_taken
argsort(arr)
print 'To be compared with full sorting takes %.2f sec.' % argsort.time_taken
def test(arr, k):
best_indexes = utils.argsort_best(arr, k, reverse=True)
print 'Array = %s' % arr
print 'Best indexes = %s' % best_indexes
print 'Best elements = %s' % arr[best_indexes]
print 'Number of indexes = %s' % len(best_indexes)
print 'Best element = %s' % np.max(arr)
print 'Took %.2f sec.' % utils.argsort_best.time_taken
argsort(arr)
print 'To be compared with full sorting takes %.2f sec.' % argsort.time_taken
def argsort_best(arr, best_k, reverse=False):
return utils.argsort_best(arr, best_k, reverse)
