Cree pequeños dataframes de pandas basados en combinaciones de valores en dos columnas.

Answer

1. 

   davy.ai

   August 5, 2023 at 11:27 am

   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ón apply para crear un nuevo dataframe para cada grupo. Dentro de la función apply, podemos utilizar la función unique para obtener los valores únicos en la columna B para el grupo actual.

   Aquí está el código para lograr esto:

   import pandas as pd
   
   df = pd.DataFrame({'A': ['a', 'a', 'a', 'b', 'b', 'b', 'c', 'c', 'c', 'd'], 'B': [1, 2, 3, 1, 2, 3, 1, 2, 3, 1]})
   
   # agrupar el dataframe por la columna A y aplicar una función para obtener los valores únicos en la columna B
   result = df.groupby('A').apply(lambda x: pd.DataFrame({'A': x['A'].unique(), 'B': x['B'].unique()}))
   

   El dataframe result resultante tendrá las siguientes filas:

   A    B
   a    1
   a    2
   a    3
   b    1
   b    2
   b    3
   c    1
   c    2
   c    3
   d    1
   
   A    B
   a    1
   a    2
   a    3
   
   A    B
   b    1
   b    2
   b    3
   
   A    B
   c    1
   c    2
   c    3
   
   A    B
   d    1
   

   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.

Comments are closed.