Welcome, guest | Sign In | My Account | Store | Cart
#!/usr/bin/env python

import numarray

def complement(ind_arr, n):
    """
    Find the complement of the set of indices in ind_arr from
    arange(n)
    """

    mat = numarray.ones(n)
    numarray.put(mat, ind_arr, 0)
    out = numarray.nonzero(mat)
    return out[0]


if __name__ == "__main__":
    orig_arr = numarray.arange(10) + 0.2
    indices = numarray.array([1, 3, 5])
    comp = complement(indices, len(orig_arr))
    comp_arr = numarray.take(orig_arr, comp)
    print "orig_arr: ", orig_arr
    print "indices: ", indices
    print "complement indices: ", comp
    print "complement elements: ", comp_arr

History

  • revision 3 (17 years ago)
  • previous revisions are not available