R中的IF語句 滿足條件時在新列中添加新元素

Question

我正在嘗試在數據框中添加一列，如果滿足該特定行的條件，則新列中的新元素的值為“1”。

為了檢查我正在遍歷另一個參考數據框的情況。

county_list = (df$county_name[df$wolves_present_in_county==1 & df$year==2015])

for (i in df$county_name) {
  for (j in county_list) {
    if (df$county_name[i]==county_list[j])
    {
      df$wolvein2015 = 1
      break
    }
  }
}

Output數據集中的錯誤

Answer 1

這是dplyr的選項

library(dplyr)
df %>%
     mutate(wolvein2015 = +((year == 2015 & !is.na(year)) &  
         as.logical(wolves_present_in_county) & !is.na(wolves_present_in_county)))

Answer 2

我認為你可以在 base R 中做你想做的事。這是一個使用 mtcars 的例子：

cars <- mtcars
cars$new <- ifelse(cars$cyl == 4 & cars$mpg > 30, 1, 0)

根據 2 個其他變量的條件，新列添加了 0/1。

順便說一句，因為 R 是一個向量化的范例，所以你應該只將for循環用作最后的手段。