[Numpy-discussion] Need help for implementing a fast clip in numpy (was slow clip)

David Cournapeau david at ar.media.kyoto-u.ac.jp
Tue Jan 9 09:19:22 EST 2007


    I am finally implementing a C function to replace the current slow 
implementation of clip in python as promised a few weeks ago. The idea 
is to implement it as the following:

def clip(input, min, max):
    a   = input.copy()
    putmask(a, a <= min, min)
    putmask(a, a >= max, max)
    return a

I don't have that much experience in writing general numpy functions, so 
I was wondering of other people could advise me on the following points.

    - should I care only about numeric types, eg something where 
PyArray_ISNUMBER returns true ?
    - If input is an array of type T1, min a scalar of type T2 and max a 
scalar of type T3, which type should be the reference, assuming they are 
all basic numeric types (float, int, etc...). Eg, if T1 is integer, T2 
and T3 are float, what am I supposed to do ? Should I use the basic rule 
as told in numpy doc, and in that case, and do something like this to 
determine common datatype (pseudo code):

    int typenum;
    for each arg in arglist:
       typenum = PyArray_ObjectType(arg, typenum)

    - PyArray_PutMask does not assume anything on the datatype of the 
input array nor this alignement, and treat its binary content as a 
buffer to char*. Do I need to do the same, or can I just use "normal" 
pointer arithmetic on arrays which elements can be casted to basic C types ?



More information about the NumPy-Discussion mailing list