CODEWITHSUNDEEP
pythonpandas
Mervyn Lee
Mervyn Lee

Reputation: 2187

Pandas conditional creation of a column value depending on other column

I have created an empty column named url. I want to set the value of the url in each row depending on the value of target. If the target is 0 then I will use the value of urlA on url for that row. Else if the target is 1 then I will use the value of urlB on the url for that row. For illustration, this is the table before.

   url   urlA   urlB    target
1         x1     y1       0
2         x2     y2       1
3         x3     y3       1
4         x4     y4       0

After:

    url   urlA   urlB    target
1   x1      x1     y1       0
2   y2      x2     y2       1
3   y3      x3     y3       1
4   x4      x4     y4       0

Upvotes: 0

Views: 27

Answers (1)

Henry Yik
Henry Yik

Reputation: 22493

Use np.where:

df["url"] = np.where(df["target"]==1, df["urlB"], df["urlA"])

#
  url urlA urlB  target
0  x1   x1   y1       0
1  y2   x2   y2       1
2  y3   x3   y3       1
3  x4   x4   y4       0

Upvotes: 2

Related Questions