Exemplos de ensure_categorical em Python

Exemplo n.º 1

Arquivo: test_inference.py Projeto: 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)

Exemplo n.º 2

Arquivo: test_inference.py Projeto: 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)

Exemplo n.º 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()

Exemplo n.º 4

Arquivo: grouper.py Projeto: 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()