Home » Python » Python Programs

Multiply two columns in a pandas dataframe and add the result into a new column

Given a DataFrame, we need to multiply two columns in this DataFrame and add the result into a new column. By Pranit Sharma Last updated : September 25, 2023

Pandas is a special tool that allows us to perform complex manipulations of data effectively and efficiently. Inside pandas, we mostly deal with a dataset in the form of DataFrame. DataFrames are 2-dimensional data structures in pandas. DataFrames consist of rows, columns, and data.

Problem statement

Given a DataFrame, we need to multiply two columns in this DataFrame and add the result into a new column.

Multiplying two columns in a pandas dataframe and add the result into a new column

For this purpose, we will first create a DataFrame, the dataframe will contain 2 columns initially and we will perform a row-wise product of both columns and store all the values in a list. Finally, we will assign this list to a new column of the DataFrame.

Let us understand with the help of an example,

Python program to multiply two columns in a pandas dataframe and add the result into a new column

# Importing pandas package
import pandas as pd

# Creating a Dictionary
d = {
    'A':[10,20,30,40,50],
    'B': [10,20,30,40,50]
}

# Creating a DataFrame
df = pd.DataFrame(d)

# Display original DataFrame
print("Original DataFrame:\n",df,"\n")

# Creating a list of values
list = []
for i in range(len(df['A'])):
    a = (df['A'][i]*df['B'][i])
    list.append(a)

# Creating a new column and assigning 
# its values as the list defined above
df['Product'] = list

# Display modified DataFrame
print("Modified DataFrame:\n",df)

Output

The output of the above program is:

Python Pandas Programs »