From 8a52d0405d8911aea31cfdfe3ef1e68f42f6081e Mon Sep 17 00:00:00 2001 From: John S Bogaardt Date: Sat, 4 Nov 2023 08:24:31 -0600 Subject: [PATCH] pandas.groupby.dtype deprecation --- chainladder/core/pandas.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/chainladder/core/pandas.py b/chainladder/core/pandas.py index 247a224d..3323973a 100644 --- a/chainladder/core/pandas.py +++ b/chainladder/core/pandas.py @@ -411,12 +411,13 @@ def aggregate(i, obj, axis, v): for i in self.groups.indices.values() ] obj = concat(values, axis=self.axis, ignore_index=True) + group_index = pd.Index(self.groups.keys()) if self.axis == 0: - if isinstance(self.groups.dtypes.index, pd.MultiIndex): + if isinstance(group_index, pd.MultiIndex): index = ( pd.DataFrame( - np.zeros(len(self.groups.dtypes.index)), - index=self.groups.dtypes.index, + np.zeros(len(group_index)), + index=group_index, columns=["_"], ) .reset_index() @@ -424,11 +425,11 @@ def aggregate(i, obj, axis, v): ) obj.index = index else: - index = pd.DataFrame(self.groups.dtypes.index) + index = pd.DataFrame(group_index) obj.key_labels = index.columns.tolist() obj.kdims = index.values if self.axis == 1: - obj.vdims = pd.DataFrame(self.groups.dtypes.index).values[:, 0] + obj.vdims = pd.DataFrame(group_index).values[:, 0] if self.axis == 2: odims = self.obj._to_datetime( pd.Series(self.groups.indices.keys()).to_frame(), [0]