Skip to main content

Create a New column

Problem Description​

A company plans to provide its employees with a bonus. Write a solution to create a new column name bonus that contains the doubled values of the salary column.

Examples​

Input:

DataFrame employees
+---------+--------+
| name    | salary |
+---------+--------+
| Piper   | 4548   |
| Grace   | 28150  |
| Georgia | 1103   |
| Willow  | 6593   |
| Finn    | 74576  |
| Thomas  | 24433  |
+---------+--------+

Output:

+---------+--------+--------+
| name    | salary | bonus  |
+---------+--------+--------+
| Piper   | 4548   | 9096   |
| Grace   | 28150  | 56300  |
| Georgia | 1103   | 2206   |
| Willow  | 6593   | 13186  |
| Finn    | 74576  | 149152 |
| Thomas  | 24433  | 48866  |
+---------+--------+--------+

Explanation:

A new column bonus is created by doubling the value in the column salary.

Intuition​

I will use the pandas library to solve this problem. I will create a new column bonus by multiplying the salary column by 2.

Solution Code​

Written by @Abhay:)
import pandas as pd

def createBonusColumn(employees: pd.DataFrame) -> pd.DataFrame:
  employees['bonus'] = employees['salary']*2
  return employees

References​