Scale Series.count() (#578)

* Scale Series.count()

* Add special case for ints which simply return len

Author: PokhodenkoSA

sdc/datatypes/hpat_pandas_series_functions.py

@@ -4551,14 +4551,22 @@ def hpat_pandas_series_count_str_impl(self, level=None):
 
         return hpat_pandas_series_count_str_impl
 
+    if isinstance(self.data, types.Array) and isinstance(self.data.dtype, types.Integer):
+        def hpat_pandas_series_count_int_impl(self, level=None):
+            return len(self._data)
+        return hpat_pandas_series_count_int_impl
+
     def hpat_pandas_series_count_impl(self, level=None):
         """
         Return number of non-NA/null observations in the object
         Returns number of unique elements in the object
         Test: python -m sdc.runtests sdc.tests.test_series.TestSeries.test_series_count
         """
-        data_no_nan = self._data[~numpy.isnan(self._data)]
-        return len(data_no_nan)
+        result = 0
+        for i in prange(len(self._data)):
+            if not numpy.isnan(self._data[i]):
+                result = result + 1
+        return result
 
     return hpat_pandas_series_count_impl
 

sdc/tests/tests_perf/test_perf_series.py

@@ -75,7 +75,7 @@ def _test_case(self, pyfunc, name, total_data_length, data_num=1, input_data=tes
        usecase_params='A, B', data_num=2),
     TC(name='copy', size=[10 ** 8]),
     TC(name='corr',  size=[10 ** 7],params='other', data_num=2),
-    TC(name='count', size=[10 ** 7]),
+    TC(name='count', size=[10 ** 8]),
     TC(name='cov', size=[10 ** 8], params='other', data_num=2),
     TC(name='cumsum', size=[10 ** 8]),
     TC(name='describe', size=[10 ** 7]),