U
|
¬ý°dA ã @ sL d dl mZ d dlZd dlmZ d dlmZmZm Z dddddd d
|
Z
|
dS ) é )Ú annotationsN)Úunique1d)Ú CategoricalÚCategoricalDtypeÚrecode_for_categoriesr Úboolz&tuple[Categorical, Categorical | None])ÚcÚsortÚobservedÚreturnc C sð |rdt | j}||dk }|r(t |¡}| j |¡}t| j| j|}t|| jd}t ||dd| fS |rp| dfS t
|
| j ¡ ¡}t | j| jdk } |r¢t | ¡} t |t | krÒtj || dd}
|
t | |
|
f¡}n| }t | | ¡ j |¡dfS )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)ÚdtypeZfastpathN)Z assume_unique)r ÚcodesÚnpr Ú
|
categoriesZtaker r r r ZarangeZnuniqueÚlenZ setdiff1dZ concatenateÚunique) r r r
|
Z unique_codesZ
|
take_codesr r r Z all_codesZunique_notnan_codesZ missing_codes© r úVd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/core/groupby/categorical.pyÚrecode_for_groupby s( $
|
|
|
r ) Ú
|
__future__r Únumpyr Zpandas.core.algorithmsr Z pandas.core.arrays.categoricalr r r r r r r r Ú<module> s
|
