Génération de fonctionnalités avec pandas group par

Je souhaite créer une quantité de fonctionnalités avec le groupe pandas en

Lorsque vous souhaitez ajouter des statistiques pour chaque attribut de colonne à une fonctionnalité, il existe de nombreux cas où vous n'avez pas besoin de créer quelque chose comme dict avec des collections ou group by et de les fusionner. C'est facile si vous ne faites que publier des statistiques, mais j'ai eu du mal à utiliser pandas.DataFrame.groupby lorsque je voulais l'ajouter à l'enregistrement en tant que fonctionnalité, je vais donc le laisser comme un mémo.

Ce que je veux dire, c'est que groupby.transform est pratique.

Exemple de données

import pandas as pd
df = pd.DataFrame({
     "site":["A","A","A","B","B","C"],
     "dat":[15,30,30,30,10,50]
})

Statistiques de base telles que moyenne, maximum et minimum pour chaque attribut

Les fonctionnalités peuvent être générées directement en modifiant l'argument de transformation en np.max ou np.min. Il en va de même pour la médiane, la var, etc. Le code pour calculer la valeur moyenne de chaque site est indiqué ci-dessous.

import numpy as np
df["site_mean"] = df.groupby("site").transform(np.mean)

Count Encoding La méthode permettant de faire du nombre d'occurrences d'entités (de catégorie) d'une certaine colonne une nouvelle caractéristique est appelée encodage de comptage. En combinant avec groupby, il est possible de caractériser quelque chose comme la rareté dans un certain attribut. Vous pouvez le faire avec collections.Counter, mais cela se termine également par une transformation.

Le code à convertir en nombre d'occurrences de la paire site et dat s'affiche. (30 apparitions sur le site A sont 2 fois)

df["count_site_dat"] = df.groupby(["site","dat"]).transform(np.size)

Classement

Calculez les plus grandes données d'une certaine caractéristique parmi les données ayant une certaine caractéristique.

df["site_rank"] = df.groupby("site")["dat"].rank(method="dense")

La modification de l'argument de rang change principalement la méthode d'expression de la même valeur (même rang). Pour plus de détails, reportez-vous à la méthode de classement par rang pandas.DataFrame, Series.