Python sort_indexの例

コード例 #1

ファイル: multiarray.py プロジェクト: Brainiarc7/afnumpy

 def argsort(self, axis=-1, kind='quicksort', order=None):
     if kind != 'quicksort':
         print( "argsort 'kind' argument ignored" )
     if order is not None:
         raise ValueError('order argument is not supported')
     if(axis < 0):
         axis = self.ndim+axis
     val, idx = arrayfire.sort_index(self.d_array, pu.c2f(self.shape, axis))
     return ndarray(self.shape, dtype=pu.typemap(idx.dtype()), af_array=idx)

コード例 #2

ファイル: multiarray.py プロジェクト: connectthefuture/afnumpy

 def argsort(self, axis=-1, kind='quicksort', order=None):
     if kind != 'quicksort':
         print("argsort 'kind' argument ignored")
     if order is not None:
         raise ValueError('order argument is not supported')
     if (axis < 0):
         axis = self.ndim + axis
     val, idx = arrayfire.sort_index(self.d_array, pu.c2f(self.shape, axis))
     return ndarray(self.shape, dtype=pu.typemap(idx.dtype()), af_array=idx)

コード例 #3

ファイル: _array.py プロジェクト: eeeedgar/model_1

def sort_argsort(a: ndarray, axis: int = -1, ascending: bool = True) \
        -> tp.Tuple[ndarray, ndarray]:
    if axis is None:
        a = a.flatten()
        axis = 0
    elif axis == -1:
        axis = a.ndim - 1
    elif axis >= a.ndim:
        raise ValueError(f"Parameter axis must be between -1 and {a.ndim - 1}")

    af_out_array, af_idx_array \
        = af.sort_index(a._af_array, dim=axis, is_ascending=ascending)

    return ndarray(af_out_array), ndarray(af_idx_array)

コード例 #4

def simple_algorithm(verbose = False):
    display_func = _util.display_func(verbose)
    print_func   = _util.print_func(verbose)

    a = af.randu(3, 3)

    print_func(af.sum(a), af.product(a), af.min(a), af.max(a),
               af.count(a), af.any_true(a), af.all_true(a))

    display_func(af.sum(a, 0))
    display_func(af.sum(a, 1))

    display_func(af.product(a, 0))
    display_func(af.product(a, 1))

    display_func(af.min(a, 0))
    display_func(af.min(a, 1))

    display_func(af.max(a, 0))
    display_func(af.max(a, 1))

    display_func(af.count(a, 0))
    display_func(af.count(a, 1))

    display_func(af.any_true(a, 0))
    display_func(af.any_true(a, 1))

    display_func(af.all_true(a, 0))
    display_func(af.all_true(a, 1))

    display_func(af.accum(a, 0))
    display_func(af.accum(a, 1))

    display_func(af.sort(a, is_ascending=True))
    display_func(af.sort(a, is_ascending=False))

    b = (a > 0.1) * a
    c = (a > 0.4) * a
    d = b / c
    print_func(af.sum(d));
    print_func(af.sum(d, nan_val=0.0));
    display_func(af.sum(d, dim=0, nan_val=0.0));

    val,idx = af.sort_index(a, is_ascending=True)
    display_func(val)
    display_func(idx)
    val,idx = af.sort_index(a, is_ascending=False)
    display_func(val)
    display_func(idx)

    b = af.randu(3,3)
    keys,vals = af.sort_by_key(a, b, is_ascending=True)
    display_func(keys)
    display_func(vals)
    keys,vals = af.sort_by_key(a, b, is_ascending=False)
    display_func(keys)
    display_func(vals)

    c = af.randu(5,1)
    d = af.randu(5,1)
    cc = af.set_unique(c, is_sorted=False)
    dd = af.set_unique(af.sort(d), is_sorted=True)
    display_func(cc)
    display_func(dd)

    display_func(af.set_union(cc, dd, is_unique=True))
    display_func(af.set_union(cc, dd, is_unique=False))

    display_func(af.set_intersect(cc, cc, is_unique=True))
    display_func(af.set_intersect(cc, cc, is_unique=False))

コード例 #5

ファイル: simple_algorithm.py プロジェクト: maxssage/arrayfire-python

af.display(af.count(a, 0))
af.display(af.count(a, 1))

af.display(af.any_true(a, 0))
af.display(af.any_true(a, 1))

af.display(af.all_true(a, 0))
af.display(af.all_true(a, 1))

af.display(af.accum(a, 0))
af.display(af.accum(a, 1))

af.display(af.sort(a, is_ascending=True))
af.display(af.sort(a, is_ascending=False))

val, idx = af.sort_index(a, is_ascending=True)
af.display(val)
af.display(idx)
val, idx = af.sort_index(a, is_ascending=False)
af.display(val)
af.display(idx)

b = af.randu(3, 3)
keys, vals = af.sort_by_key(a, b, is_ascending=True)
af.display(keys)
af.display(vals)
keys, vals = af.sort_by_key(a, b, is_ascending=False)
af.display(keys)
af.display(vals)

c = af.randu(5, 1)

コード例 #6

ファイル: simple_algorithm.py プロジェクト: nagyistoce/arrayfire_python

af.print_array(af.count(a, 0))
af.print_array(af.count(a, 1))

af.print_array(af.any_true(a, 0))
af.print_array(af.any_true(a, 1))

af.print_array(af.all_true(a, 0))
af.print_array(af.all_true(a, 1))

af.print_array(af.accum(a, 0))
af.print_array(af.accum(a, 1))

af.print_array(af.sort(a, is_ascending=True))
af.print_array(af.sort(a, is_ascending=False))

val,idx = af.sort_index(a, is_ascending=True)
af.print_array(val)
af.print_array(idx)
val,idx = af.sort_index(a, is_ascending=False)
af.print_array(val)
af.print_array(idx)

b = af.randu(3,3)
keys,vals = af.sort_by_key(a, b, is_ascending=True)
af.print_array(keys)
af.print_array(vals)
keys,vals = af.sort_by_key(a, b, is_ascending=False)
af.print_array(keys)
af.print_array(vals)

c = af.randu(5,1)

コード例 #7

ファイル: algorithm.py プロジェクト: roaffix/arrayfire-python

def simple_algorithm(verbose=False):
    display_func = _util.display_func(verbose)
    print_func = _util.print_func(verbose)

    a = af.randu(3, 3)
    k = af.constant(1, 3, 3, dtype=af.Dtype.u32)
    af.eval(k)

    print_func(af.sum(a), af.product(a), af.min(a), af.max(a), af.count(a),
               af.any_true(a), af.all_true(a))

    display_func(af.sum(a, 0))
    display_func(af.sum(a, 1))

    rk = af.constant(1, 3, dtype=af.Dtype.u32)
    rk[2] = 0
    af.eval(rk)
    display_func(af.sumByKey(rk, a, dim=0))
    display_func(af.sumByKey(rk, a, dim=1))

    display_func(af.productByKey(rk, a, dim=0))
    display_func(af.productByKey(rk, a, dim=1))

    display_func(af.minByKey(rk, a, dim=0))
    display_func(af.minByKey(rk, a, dim=1))

    display_func(af.maxByKey(rk, a, dim=0))
    display_func(af.maxByKey(rk, a, dim=1))

    display_func(af.anyTrueByKey(rk, a, dim=0))
    display_func(af.anyTrueByKey(rk, a, dim=1))

    display_func(af.allTrueByKey(rk, a, dim=0))
    display_func(af.allTrueByKey(rk, a, dim=1))

    display_func(af.countByKey(rk, a, dim=0))
    display_func(af.countByKey(rk, a, dim=1))

    display_func(af.product(a, 0))
    display_func(af.product(a, 1))

    display_func(af.min(a, 0))
    display_func(af.min(a, 1))

    display_func(af.max(a, 0))
    display_func(af.max(a, 1))

    display_func(af.count(a, 0))
    display_func(af.count(a, 1))

    display_func(af.any_true(a, 0))
    display_func(af.any_true(a, 1))

    display_func(af.all_true(a, 0))
    display_func(af.all_true(a, 1))

    display_func(af.accum(a, 0))
    display_func(af.accum(a, 1))

    display_func(af.scan(a, 0, af.BINARYOP.ADD))
    display_func(af.scan(a, 1, af.BINARYOP.MAX))

    display_func(af.scan_by_key(k, a, 0, af.BINARYOP.ADD))
    display_func(af.scan_by_key(k, a, 1, af.BINARYOP.MAX))

    display_func(af.sort(a, is_ascending=True))
    display_func(af.sort(a, is_ascending=False))

    b = (a > 0.1) * a
    c = (a > 0.4) * a
    d = b / c
    print_func(af.sum(d))
    print_func(af.sum(d, nan_val=0.0))
    display_func(af.sum(d, dim=0, nan_val=0.0))

    val, idx = af.sort_index(a, is_ascending=True)
    display_func(val)
    display_func(idx)
    val, idx = af.sort_index(a, is_ascending=False)
    display_func(val)
    display_func(idx)

    b = af.randu(3, 3)
    keys, vals = af.sort_by_key(a, b, is_ascending=True)
    display_func(keys)
    display_func(vals)
    keys, vals = af.sort_by_key(a, b, is_ascending=False)
    display_func(keys)
    display_func(vals)

    c = af.randu(5, 1)
    d = af.randu(5, 1)
    cc = af.set_unique(c, is_sorted=False)
    dd = af.set_unique(af.sort(d), is_sorted=True)
    display_func(cc)
    display_func(dd)

    display_func(af.set_union(cc, dd, is_unique=True))
    display_func(af.set_union(cc, dd, is_unique=False))

    display_func(af.set_intersect(cc, cc, is_unique=True))
    display_func(af.set_intersect(cc, cc, is_unique=False))

コード例 #8

ファイル: helloworld.py プロジェクト: arrayfire/arrayfire-python

    af.display(C);

    print("Grab last row\n");
    c = C[-1,:];
    af.display(c);

    print("Scan Test\n");
    r = af.constant(2, 16, 4, 1, 1);
    af.display(r);

    print("Scan\n");
    S = af.scan(r, 0, af.BINARYOP.MUL);
    af.display(S);

    print("Create 2-by-3 matrix from host data\n");
    d = [ 1, 2, 3, 4, 5, 6 ]
    D = af.Array(d, (2, 3))
    af.display(D)

    print("Copy last column onto first\n");
    D[:,0] = D[:, -1]
    af.display(D);

    print("Sort A and print sorted array and corresponding indices\n");
    [sorted_vals, sorted_idxs] = af.sort_index(A);
    af.display(A)
    af.display(sorted_vals)
    af.display(sorted_idxs)
except Exception as e:
    print("Error: " + str(e))