matrix.py 文件源码

def get_membership_mask(self, labels, rows_or_columns):
        from .util import array_tostring

        assert rows_or_columns in ['rows', 'columns']
        assert isinstance(labels, np.ndarray)
        assert labels.size > 0

        if rows_or_columns == "rows":
            filter_labels = self.rowlabels
        else:
            filter_labels = self.columnlabels

        labels = array_tostring(labels)
        filter_labels = array_tostring(filter_labels)

        return np.in1d(filter_labels.ravel(),
                       labels).reshape(filter_labels.shape)