与 ifelse 函数的关联错误

Question

>我有一个大数据框架，需要创建一个带有"月"列的分类变量"季节"的列。

structure(list(year = c("2017", "2017", "2017", "2017", "2017", 
"2017", "2017", "2017", "2017", "2017", "2017", "2017", "2017", 
"2017", "2017", "2017", "2017", "2017", "2017", "2017", "2017", 
"2017", "2017", "2017", "2017", "2017", "2017", "2017", "2017", 
"2017", "2017", "2017", "2017", "2017", "2017", "2017", "2017", 
"2017", "2017", "2017", "2017", "2017", "2017", "2017", "2017", 
"2017", "2017", "2017", "2017", "2017"), month = c(1, 2, 2, 1, 
1, 3, 3, 3, 3, 4, 4, 4, 4, 4, 3, 1, 3, 3, 4, 5, 1, 2, 2, 2, 2, 
3, 1, 1, 2, 3, 4, 5, 6, 2, 5, 8, 1, 1, 4, 2, 3, 4, 2, 2, 2, 3, 
3, 4, 4, 1), day = c("29", "12", "12", "25", "25", "01", "01", 
"29", "29", "10", "10", "10", "10", "10", "14", "31", "02", "28", 
"25", "31", "21", "18", "12", "01", "01", "28", "07", "18", "16", 
"30", "26", "24", "22", "12", "16", "13", "10", "10", "11", "01", 
"28", "29", "04", "01", "01", "28", "28", "29", "29", "10")), row.names = c(NA, 
-50L), class = c("tbl_df", "tbl", "data.frame"))

我使用了像下面这样的ifelse函数：

a <- c(3,4,5)
b <- c(6,7,8)
c <- c(9,10,11)
d <- c(12,1,2)
df$season <- ifelse(df$month == a, "Spring",
ifelse(df$month == b, "Summer",
ifelse(df$month == c, "Fall",
ifelse(df$month == d, "Winter",
""))))

但是我在"季节"列中得到了错误的联想。许多应该有季节名称的值是" "。

我也试过：

df[df$month == a, ][, "Season"] <- "Spring"
df[df$month == b, ][, "Season"] <- "Summer"
df[df$month == c, ][, "Season"] <- "Fall"
df[df$month == d, ][, "Season"] <- "Winter"

但它没有奏效。

Answer 1

这里的问题是每个vector元素都属于length> 1。 为此，我们可以使用%in%

df$season <- ifelse(df$month %in% a, "Spring",
ifelse(df$month %in% b, "Summer",
ifelse(df$month %in% c, "Fall",
ifelse(df$month %in% d, "Winter",
""))))

---

另外，在tidyverse，我们可以做

df %>%
mutate(season = case_when(month %in% a ~ "Spring",
month %in% b ~ "Summer",
month %in% c ~ "Fall",
month %in% dd ~ "Winter",
TRUE ~ NA_character_))

---

如果有很多值，更好的方法是在创建键/值数据集后连接

library(tidyverse)
map2_df(lst(a, b, c, d), c("Spring", "Summer", "Fall", "Winter"),
~ tibble(month = .x, val = .y)) %>% 
right_join(df) %>% 
select(month = val, year, day)

Answer 2

你应该做：

df$season <- ifelse(df$month %in% a, "Spring",
ifelse(df$month %in% b, "Summer",
ifelse(df$month %in% c, "Fall",
ifelse(df$month %in% d, "Winter",
""))))

由于您使用的是"=="，因此您仅将df$month与a，b，c和d的第一个元素进行比较。