Create a pandas column with values from filtering string values from another column

Question

I have a dataframe as:

Feature     Prediction 
Dell 12-4.  Grade 1
Dell 12-5.  Grade 1
Dell 10-4.  Grade 1
Dell 10-5.  Grade 0
Dell 12-6.  Grade 0

I would like to create a new column called 'True Outcome', which filters Feature Column for strings that starts with 'Dell 12-' and assigns Grade 0 to it otherwise Grade 1.

The expected output should look like this:

Feature     Prediction  True Outcome
Dell 12-4.  Grade 1.    Grade 0
Dell 12-5.  Grade 1.    Grade 0
Dell 10-4.  Grade 1.    Grade 1
Dell 10-5.  Grade 0.    Grade 1
Dell 12-6.  Grade 0.    Grade 0

Ami Tavory · Accepted Answer · 2021-06-30 10:50:30Z

2

To find if the column starts with the requested string, you can use

df.Feature.str.startswith('Dell 12-')

and the assignment can be

import numpy as np
df['Outcome'] = np.where(df.Feature.str.startswith('Dell 12-'), 'Grade 0', 'Grade 1')

edited Jun 30, 2021 at 10:50

answered Jun 30, 2021 at 8:45

Ami Tavory

76.7k13 gold badges152 silver badges196 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

CutePoison · Accepted Answer · 2021-06-30 08:46:51Z

0

First create a column with all 1s - then set all the values where feature contains Dell-12

df["True Outcome"] = 1  #Create column
df.loc[df["features"].str.startswith("Dell 12") ,"True Outcome"]=0 #Set the values to 0

answered Jun 30, 2021 at 8:46

CutePoison

5,5648 gold badges44 silver badges100 bronze badges

Comments

StackUser · Accepted Answer · 2021-06-30 08:55:12Z

0

A better approach:

df['True Outcome'] = np.where(df['Feature'].str.startswith('Dell 12'), 'Grade 0', 'Grade 1')

answered Jun 30, 2021 at 8:55

StackUser

4851 gold badge5 silver badges16 bronze badges

Collectives™ on Stack Overflow

Create a pandas column with values from filtering string values from another column

3 Answers 3

Comments

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

Comments

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related