Skip to content
This repository was archived by the owner on Feb 2, 2024. It is now read-only.

Commit 8216475

Browse files
Rubtsowashssf
authored andcommitted
fix nunique (#341)
* fix nunique * fix Series.nunique
1 parent 6401f3d commit 8216475

1 file changed

Lines changed: 6 additions & 5 deletions

File tree

sdc/datatypes/hpat_pandas_series_functions.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2896,11 +2896,12 @@ def hpat_pandas_series_nunique_str_impl(self, dropna=True):
28962896
It is better to merge with Numeric branch
28972897
"""
28982898

2899-
str_set = set(self._data)
2900-
if dropna == False:
2901-
return len(str_set) - 1
2902-
else:
2903-
return len(str_set)
2899+
data = self._data
2900+
if dropna:
2901+
nan_mask = self.isna()
2902+
data = self._data[~nan_mask._data]
2903+
unique_values = set(data)
2904+
return len(unique_values)
29042905

29052906
return hpat_pandas_series_nunique_str_impl
29062907

0 commit comments

Comments
 (0)