From eab9f9fdc1d24defa846742475e865fd97c01949 Mon Sep 17 00:00:00 2001 From: Wolf Behrenhoff Date: Wed, 31 May 2023 16:11:49 +0200 Subject: [PATCH] TYP: Fix typing issues, Label => IndexLabel --- pandas/core/frame.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/pandas/core/frame.py b/pandas/core/frame.py index 3e6c89139d06d..ad1edc1a675e5 100644 --- a/pandas/core/frame.py +++ b/pandas/core/frame.py @@ -6956,7 +6956,7 @@ def sort_index( def value_counts( self, - subset: Sequence[Hashable] | None = None, + subset: IndexLabel | None = None, normalize: bool = False, sort: bool = True, ascending: bool = False, @@ -7080,8 +7080,8 @@ def value_counts( if normalize: counts /= counts.sum() - # Force MultiIndex for single column - if is_list_like(subset) and len(subset) == 1: + # Force MultiIndex for a list_like subset with a single column + if is_list_like(subset) and len(subset) == 1: # type: ignore[arg-type] counts.index = MultiIndex.from_arrays( [counts.index], names=[counts.index.name] ) @@ -8991,7 +8991,7 @@ def pivot_table( sort=sort, ) - def stack(self, level: Level = -1, dropna: bool = True, sort: bool = True): + def stack(self, level: IndexLabel = -1, dropna: bool = True, sort: bool = True): """ Stack the prescribed level(s) from columns to index. @@ -9296,7 +9296,7 @@ def explode( return result.__finalize__(self, method="explode") - def unstack(self, level: Level = -1, fill_value=None, sort: bool = True): + def unstack(self, level: IndexLabel = -1, fill_value=None, sort: bool = True): """ Pivot a level of the (necessarily hierarchical) index labels.