Pandas – All combinations of two columns
In this article, we will see how to get the combination of two columns of a DataFrame. First, let’s create a sample DataFrame.
Code: An example code to create a data frame using dictionary.
Python3
# importing pandas module for the # data frameimport pandas as pd # creating data frame for student details # using dictionarydata = pd.DataFrame({'id': [7058, 7059, ], 'name': ['sravan', 'jyothika']}) print(data) |
Output:
To combine two columns in a data frame using itertools module. It provides various functions that work on iterators to produce complex iterators. To get all combinations of columns we will be using itertools.product module. This function computes the cartesian product of input iterables. To compute the product of an iterable with itself, we use the optional repeat keyword argument to specify the number of repetitions. The output of this function is tuples in sorted order.
Syntax: itertools.product(iterables, repeat=1)
Code:
Python3
# import pandas as pdimport pandas as pd # creating data framedf = pd.DataFrame(data=[['sravan', 'Sudheer'], ['radha', 'vani'], ], columns=['gents', 'ladies']) print(df) |
Output:
Code:
Python3
# importing productfrom itertools import product # apply product methodprint(list(product(df['gents'], df['ladies']))) |
Output:
