add Int8Index

Author: tp

pandas/_libs/algos_common_helper.pxi.in

@@ -25,9 +25,10 @@ WARNING: DO NOT edit .pxi FILE directly, .pxi is generated from .pxi.in
 dtypes = [('float64', 'float64_t', 'np.float64', True, True),
           ('float32', 'float32_t', 'np.float32', True, True),
           ('object', 'object', 'object', True, False),
-          ('int32', 'int32_t', 'np.int32', False, True),
           ('int64', 'int64_t', 'np.int64', False, True),
           ('uint64', 'uint64_t', 'np.uint64', False, True),
+          ('int32', 'int32_t', 'np.int32', False, True),
+          ('int8', 'int8_t', 'np.int8', False, True),
           ('bool', 'uint8_t', 'np.bool', False, True)]
 
 def get_dispatch(dtypes):

pandas/_libs/index.pyx

@@ -8,8 +8,10 @@ from cpython.slice cimport PySlice_Check
 
 import numpy as np
 cimport numpy as cnp
-from numpy cimport (ndarray, float64_t, int32_t,
-                    int64_t, uint8_t, uint64_t, intp_t)
+from numpy cimport (ndarray, float64_t,
+                    int8_t, int16_t, int32_t, int64_t,
+                    uint8_t, uint64_t,
+                    intp_t)
 cnp.import_array()
 
 cdef extern from "numpy/arrayobject.h":

pandas/_libs/index_class_helper.pxi.in

@@ -12,9 +12,13 @@ WARNING: DO NOT edit .pxi FILE directly, .pxi is generated from .pxi.in
 
 # name, dtype, ctype
 dtypes = [('Float64', 'float64', 'float64_t'),
-          ('UInt64', 'uint64', 'uint64_t'),
           ('Int64', 'int64', 'int64_t'),
-          ('Object', 'object', 'object')]
+          ('Int32', 'int32', 'int32_t'),
+          ('Int16', 'int16', 'int16_t'),
+          ('Int8', 'int8', 'int8_t'),
+          ('UInt64', 'uint64', 'uint64_t'),
+          ('Object', 'object', 'object'),
+          ]
 }}
 
 {{for name, dtype, ctype in dtypes}}

pandas/core/indexes/category.py

@@ -72,7 +72,11 @@ class CategoricalIndex(Index, accessor.PandasDelegate):
     """
 
     _typ = 'categoricalindex'
-    _engine_type = libindex.Int64Engine
+
+    @property
+    def _engine_type(self):
+        type_name = self.codes.dtype.name.capitalize()
+        return getattr(libindex, "{}Engine".format(type_name))
     _attributes = ['name']
 
     def __new__(cls, data=None, categories=None, ordered=None, dtype=None,
@@ -366,7 +370,7 @@ def argsort(self, *args, **kwargs):
     def _engine(self):
 
         # we are going to look things up with the codes themselves
-        return self._engine_type(lambda: self.codes.astype('i8'), len(self))
+        return self._engine_type(lambda: self.codes, len(self))
 
     # introspection
     @cache_readonly