How do I add a new column to a dataframe in Julia using conditional logic?

General Usage
,
1

I want to create a new column in my dataframe. If another column equals zero, I want this column to equal zero. If it doesn’t, I would like this column to equal another column divided by a third column. I tried using the @rtransform dataframes meta macro but keep getting various errors. Is there a generally accepted way to do this?

2 
@rtransform df :z = :y == 0 ? 0 : :z
1 Like
3

It’s true that it might be nice for a more convenient @replace command because there is no way to do this without repeating the variable name more frequently than is ideal.

4

Base Julia:

julia> df = DataFrame(a = Int.(rand(Bool, 10)), b = 1:10, c = 10:-1:1)

julia> df.d = @. ifelse(df.a == 0, 0.0, df.b / df.c); df
10×4 DataFrame
 Row │ a      b      c      d        
     │ Int64  Int64  Int64  Float64  
─────┼───────────────────────────────
   1 │     0      1     10   0.0
   2 │     0      2      9   0.0
   3 │     1      3      8   0.375
   4 │     0      4      7   0.0
   5 │     0      5      6   0.0
   6 │     0      6      5   0.0
   7 │     1      7      4   1.75
   8 │     1      8      3   2.66667
   9 │     1      9      2   4.5
  10 │     1     10      1  10.0
5 Likes
5

You could calculate/construct the new column, by putting into formula the conditions you have illustrated

df.d.=(df.b./df.c).*(df.a.!=0)
#or 
@. df.d=(df.b/df.c)*(df.a!=0)
2 Likes
6

This worked perfectly. Thank you @pdeffebach !