Pandas-两列的所有组合

在这篇文章中，我们将看到如何获得一个DataFrame的两列的组合。首先，让我们创建一个样本的DataFrame。

代码：使用字典创建一个数据框架的示例代码。

# 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)

输出:

使用itertools模块来组合数据框架中的两列。它提供了各种对迭代器工作的函数，以产生复杂的迭代器。为了得到所有列的组合，我们将使用itertools.product模块。这个函数计算输入迭代器的笛卡尔乘积。为了计算一个迭代器与自身的乘积，我们使用可选的重复关键字参数来指定重复的次数。这个函数的输出是按排序顺序排列的图元。

语法:

itertools.product(iterables, repeat=1)

代码:

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

输出:

代码:

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

输出: