我的数据框如下:
S A B C D E
1 N N N N N
2 N Y Y N N
3 Y N Y N N
4 Y N Y Y Y
我需要在哪里创建一个新列F,其中包含A,B,C,D和E多个列中出现次数最多的字符?
输出应如下所示:
S A B C D E F
1 N N N N N N
2 N Y Y N N N
3 Y N Y N N N
4 Y N Y Y Y Y
答案 0 :(得分:5)
我们可以创建一个Mode
函数并将其应用于行
df1$F <- apply(df1[-1], 1, Mode)
df1
# S A B C D E F
#1 1 N N N N N N
#2 2 N Y Y N N N
#3 3 Y N Y N N N
#4 4 Y N Y Y Y Y
或者另一个选择是
df1$F <- c('N', 'Y')[max.col(table(c(row(df1[-1])), unlist(df1[-1])), 'first')]
其中
Mode <- function(x) {
ux <- unique(x)
ux[which.max(tabulate(match(x, ux)))]
}
或使用tidyverse
library(tidyverse)
df1 %>%
mutate(F = pmap_chr(.[-1], ~ Mode(c(...))))
或者另一个选择是
gather(df1, key, F, - S) %>%
group_by(S, F) %>%
summarise(n = n()) %>%
slice(which.max(n)) %>%
ungroup %>%
dplyr::select(F) %>%
bind_cols(df1, .)
或者我们转置数据集,对每列应用Mode
,然后将输出作为新列绑定到原始数据集
t(df1[-1]) %>%
as.data.frame %>%
summarise_all(Mode) %>%
unlist %>%
bind_cols(df1, F = .)
或带有data.table
的选项
library(data.table)
setDT(df1)[, F := names(which.max(table(unlist(.SD)))), S][]
注意:这些是常规方法,而不是仅检查单个案例
如果我们需要一种高效的方法而没有任何ifelse
,我们也可以通过
df1$F <- c("Y", "N")[(rowSums(df1[-1] == "N") > 2) + 1]
df1$F
#[1] "N" "N" "N" "Y"
或与Reduce
c("Y", "N")[(Reduce(`+`, lapply(df1[-1], `==`, "N")) > 2) + 1]
或者另一种方法是
c("Y", "N")[(str_count(do.call(paste0, df1[-1]), "N") > 2) + 1]
df1 <- structure(list(S = 1:4, A = c("N", "N", "Y", "Y"), B = c("N",
"Y", "N", "N"), C = c("N", "Y", "Y", "Y"), D = c("N", "N", "N",
"Y"), E = c("N", "N", "N", "Y")), class = "data.frame", row.names = c(NA,
-4L))
答案 1 :(得分:3)
一种dplyr
可能是:
df %>%
mutate(F = ifelse(rowSums(.[2:length(.)] == "N") > 2, "N", "Y"))
S A B C D E F
1 1 N N N N N N
2 2 N Y Y N N N
3 3 Y N Y N N N
4 4 Y N Y Y Y Y
假定只有N
和Y
值,并且列数为5。
正如@Sotos指出的那样,可以很容易地将其重写为base R
形式:
df$F <- ifelse(rowSums(df[2:length(df)] == "N") > 2, "N", "Y")
或者不考虑列数(基于@TinglTanglBob):
df %>%
mutate(F = ifelse(rowMeans(.[2:length(.)] == "N") > 0.5, "N", "Y"))
与基数R相同:
df$F <- ifelse(rowMeans(df[2:length(df)] == "N") > 0.5, "N", "Y")
答案 2 :(得分:2)
另一种选择,略有不同:
x$F <- unlist(do.call(Map, c(function(...) names(sort(-table(c(...)), partial=1)[1]), x[,-1])))
x
# S A B C D E F
# 1 1 N N N N N N
# 2 2 N Y Y N N N
# 3 3 Y N Y N N N
# 4 4 Y N Y Y Y Y
也许我现在只是想产生晦涩的代码...
我意识到这可能比绝对必要更笼统。无论行之间存在多少不同的事物,这都会找到最频繁的“事物”。
sort(..., partial=1)
在第一次通过后停止排序。
答案 3 :(得分:1)
或者:
d <- read.table(text ="S A B C D E
1 N N N N N
2 N Y Y N N
3 Y N Y N N
4 Y N Y Y Y", header = TRUE, row.names = 1, stringsAsFactors = FALSE)
d$F <- with(
stack(data.frame(t(as.matrix(d)), stringsAsFactors = FALSE)),
tapply(values, ind, function(x) names(sort(table(x), decreasing = TRUE)[1])))
d
#A B C D E F
#1 N N N N N N
#2 N Y Y N N N
#3 Y N Y N N N
#4 Y N Y Y Y Y