基于其他列添加列

Question

If I have a dataset called data.

I need to add a column based on this logic 0 if "gender" in data is 0 or 1 or 3; 1 if "gender" in data is 2

How can I create a code to add in?

Answer 1

Assuming gender would only take on the possible values 0, 1, and 2, we can try using ifelse as follows:

data$flag <- ifelse(data$gender <= 1, 0, 1)

We can use case_when() from the dplyr library for finer grain control:

data$flag <- case_when(
    gender %in% c(0, 1) ~ 0,
    gender == 2 ~ 1,
    # add other mappings here
    TRUE ~ NA
)