Pandas-两列的所有组合

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

输出:

Pandas--两列的所有组合

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

语法:

itertools.product(iterables, repeat=1)
Python

代码:

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

输出:

Pandas--两列的所有组合

代码:

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

输出:

Pandas--两列的所有组合

Python教程

Java教程

Web教程

数据库教程

图形图像教程

大数据教程

开发工具教程

计算机教程

登录

注册