Python ensure_categorical Examples

Example #1

File: test_inference.py Project: yyonggonyy/pandas

def test_ensure_categorical():
    values = np.arange(10, dtype=np.int32)
    result = ensure_categorical(values)
    assert (result.dtype == 'category')

    values = Categorical(values)
    result = ensure_categorical(values)
    tm.assert_categorical_equal(result, values)

Example #2

File: test_inference.py Project: josham/pandas

def test_ensure_categorical():
    values = np.arange(10, dtype=np.int32)
    result = ensure_categorical(values)
    assert (result.dtype == 'category')

    values = Categorical(values)
    result = ensure_categorical(values)
    tm.assert_categorical_equal(result, values)

Example #3

    def indices(self):
        # we have a list of groupers
        if isinstance(self.grouper, BaseGrouper):
            return self.grouper.indices

        values = ensure_categorical(self.grouper)
        return values._reverse_indexer()

Example #4

File: grouper.py Project: forking-repos/pandas

    def indices(self):
        # we have a list of groupers
        if isinstance(self.grouper, BaseGrouper):
            return self.grouper.indices

        values = ensure_categorical(self.grouper)
        return values._reverse_indexer()