Pandas: todas las combinaciones de dos columnas

En este artículo veremos cómo obtener la combinación de dos columnas de un DataFrame. Primero, creemos un DataFrame de muestra.

Código: un código de ejemplo para crear un marco de datos utilizando el diccionario.

# importing pandas module for the  
# data frame
import pandas as pd
  
# creating data frame for student details 
# using dictionary
data = pd.DataFrame({'id': [7058, 7059, ], 
                     'name': ['sravan', 'jyothika']})
  
print(data)

Producción:

Para combinar dos columnas en un marco de datos usando el módulo itertools . Proporciona varias funciones que funcionan en iteradores para producir iteradores complejos. Para obtener todas las combinaciones de columnas, utilizaremos el módulo itertools.product . Esta función calcula el producto cartesiano de iterables de entrada. Para calcular el producto de un iterable consigo mismo, usamos el argumento de palabra clave de repetición opcional para especificar el número de repeticiones. La salida de esta función son tuplas ordenadas.

Sintaxis: itertools.product(iterables, repetir=1)

Código:

# import pandas as pd
import pandas as pd
  
# creating data frame
df = pd.DataFrame(data=[['sravan', 'Sudheer'], 
                        ['radha', 'vani'], ],
                  columns=['gents', 'ladies'])
  
print(df)

Producción:

Código:

# importing product
from itertools import product
  
# apply product method
print(list(product(df['gents'], df['ladies'])))

Producción: