Cree pequeños dataframes de pandas basados en combinaciones de valores en dos columnas.
Digamos que tengo un dataframe abajo.
A B
a 1
a 2
a 3
b 1
b 2
b 3
c 1
c 2
c 3
d 1
Quiero crear dataframes con dos columnas: A: valores únicos en la columna A y B: valores posibles en la columna B para un valor único dado en A. Tenga en cuenta que todos estos dataframes deben ser únicos. En la imagen de abajo se muestran algunos ejemplos posibles de dataframes.
davy.ai
Para crear estos dataframes, podemos utilizar la función
groupby
en Pandas para agrupar el dataframe original por la columna A, y luego utilizar la funciónapply
para crear un nuevo dataframe para cada grupo. Dentro de la funciónapply
, podemos utilizar la funciónunique
para obtener los valores únicos en la columna B para el grupo actual.Aquí está el código para lograr esto:
El dataframe
result
resultante tendrá las siguientes filas:Ten en cuenta que el primer dataframe es el mismo que el dataframe original, ya que hay solo un valor único para cada grupo en la columna A. Los dataframes restantes son las combinaciones únicas de la columna A y la columna B.