o 5c%@shddlmZddlmZddlZddlmZddlm Z m Z m Z er(ddl m Z dddZdddZdS)) annotations) TYPE_CHECKINGN)unique1d) CategoricalCategoricalDtyperecode_for_categories)CategoricalIndexcrsortboolobservedreturn&tuple[Categorical, Categorical | None]c Cs|r3t|j}||dk}|jrt|}|j|}t|j|j|}t||jd}t ||dd|fS|r9|dfS| }|j|jdk}|jrMt|}| |j|}| |j|j |j}||jdfS)a= Code the categories to ensure we can groupby for categoricals. If observed=True, we return a new Categorical with the observed categories only. If sort=False, return a copy of self, coded with categories as returned by .unique(), followed by any categories not appearing in the data. If sort=True, return self. This method is needed solely to ensure the categorical index of the GroupBy result has categories in the order of appearance in the data (GH-8868). Parameters ---------- c : Categorical sort : bool The value of the sort parameter groupby was called with. observed : bool Account only for the observed values Returns ------- Categorical If sort=False, the new categories are set to the order of appearance in codes (unless ordered=True, in which case the original order is preserved), followed by any unrepresented categories in the original order. Categorical or None If we are observed, return the original categorical, otherwise None )orderedT)dtypefastpathN)rcodesrnpr categoriestakerrruniqueset_categoriesadd_categoriesisinreorder_categories) r r r unique_codes take_codesrrrcatrU/var/www/html/gps/gps/lib/python3.10/site-packages/pandas/core/groupby/categorical.pyrecode_for_groupbys$$     r!circCs0|r||jS|j|j|j}||S)a; Reverse the codes_to_groupby to account for sort / observed. Parameters ---------- c : Categorical sort : bool The value of the sort parameter groupby was called with. ci : CategoricalIndex The codes / categories to recode Returns ------- CategoricalIndex )rrrr)r r r"new_catsrrr recode_from_groupby_s  r$)r rr r r r r r)r rr r r"rr r) __future__rtypingrnumpyrpandas.core.algorithmsrpandas.core.arrays.categoricalrrrpandas.core.indexes.apirr!r$rrrr s    M