bug fix with group counts

This commit is contained in:
Steve Nyemba 2022-12-09 16:20:17 -06:00
parent 7f8754b5f1
commit 17cb39acd7
2 changed files with 3 additions and 3 deletions

View File

@ -69,7 +69,7 @@ class deid :
if _columns : if _columns :
self._compute = Compute(sample = _sample,columns=_columns) self._compute = Compute(sample = _sample,columns=_columns)
else: else:
self._comput = Compute(sample=_sample) self._compute = Compute(sample=_sample)
self._pcompute= Population() self._pcompute= Population()
def explore(self,**args): def explore(self,**args):
@ -354,7 +354,7 @@ class Compute(Risk):
self.cache['count']['fields'] = len(self._columns) self.cache['count']['fields'] = len(self._columns)
if self._sample.shape[0] > 0 and self._columns: if self._sample.shape[0] > 0 and self._columns:
_sample = _args ['sample'] _sample = _args ['sample']
_groups = self._sample.groupby(self._columns,as_index=False).count().values _groups = self._sample.groupby(self._columns,as_index=False).size().values
self.set('groups',_groups) self.set('groups',_groups)
self.cache['count']['groups'] = len(_groups) self.cache['count']['groups'] = len(_groups)

View File

@ -5,7 +5,7 @@ from setuptools import setup, find_packages
setup( setup(
name = "privacykit", name = "privacykit",
version = "0.9.0", version = "0.9.2",
author = "Healthcare/IO - The Phi Technology LLC & Health Information Privacy Lab", author = "Healthcare/IO - The Phi Technology LLC & Health Information Privacy Lab",
author_email = "info@the-phi.com", author_email = "info@the-phi.com",
license = "MIT", license = "MIT",