Pandas – 两列的所有组合

在本文中，我们将看到如何获取 DataFrame 的两列组合。首先，让我们创建一个示例 DataFrame。

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

Python3

# 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模块。此函数计算输入迭代的笛卡尔积。为了计算可迭代对象与自身的乘积，我们使用可选的 repeat 关键字参数来指定重复次数。此函数的输出是按排序顺序的元组。

Syntax: itertools.product(iterables, repeat=1)

代码：

蟒蛇3

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

输出：

代码：

蟒蛇3

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

输出：