numpy.np_in1d

Here are the examples of the python api numpy.np_in1d taken from open source projects. By voting up you can indicate which examples are most useful and appropriate.

1 Examples 7

Example 1

Project: mpop Source File: nwcsaf_hrw_hdf.py
    def filter(self, **kwargs):

        # if empty than return self (already empty)
        if self.channel.size == 0:
            return self

        HRW_new = deepcopy(self)

        for key_filter in ['min_correlation', 'min_conf_nwp', 'min_conf_no_nwp', 'cloud_type', 'level']:
            if key_filter in kwargs.keys():
                
                # if argument given is None or all keyword than skip this filter 
                if kwargs[key_filter] == None or kwargs[key_filter] == 'all' or kwargs[key_filter] == 'ALL' or kwargs[key_filter] == 'A':
                    continue

                n1 = str(HRW_new.channel.size)

                if key_filter == 'min_correlation':
                    inds = np_where(HRW_new.correlation > kwargs[key_filter])
                elif key_filter == 'min_conf_nwp':
                    inds = np_where(HRW_new.conf_nwp    > kwargs[key_filter])
                elif key_filter == 'min_conf_no_nwp':
                    inds = np_where(HRW_new.conf_no_nwp > kwargs[key_filter])
                elif key_filter == 'cloud_type':
                    mask = np_in1d(HRW_new.cloud_type, kwargs[key_filter]) 
                    inds = np_where(mask)[0]
                elif key_filter == 'level':
                    if kwargs[key_filter] == 'H': # high level: < 440hPa like in the ISCCP
                        inds = np_where(HRW_new.pressure < 44000 ) 
                    elif kwargs[key_filter] == 'M': # mid level: 440hPa ... 680hPa like in the ISCCP
                        inds = np_where( np_logical_and(44000 < HRW_new.pressure, HRW_new.pressure < 68000) ) 
                    elif kwargs[key_filter] == 'L': # low level: > 680hPa like in the ISCCP
                        inds = np_where(68000 < HRW_new.pressure)

                HRW_new.subset(inds)
                print "    filter for "+key_filter+" = ", kwargs[key_filter],' ('+n1+'->'+str(HRW_new.channel.size)+')'

        return HRW_new