作为输入dataFrame,具有这样的数据结构:
dframe <- data.frame(id = c(1,2), col1 = c(4.2, 1.2), su = c(1.2,2.4))
如何比较两列的每一行中的数字,并且根据最大的产生一个新列,该列将具有最大值的col名称和带有此值的第二列?
dcom <- data.frame(id = c(1,2), col1 = c(4.2, 1.2), su = c(1.2,2.4), com = c("col1", "su"), com_num = c(4.2, 2.4))
dcom
id col1 su com com_num
1 1 4.2 1.2 col1 4.2
2 2 1.2 2.4 su 2.4
矢量化解决方案可以是以下内容。
com_num <- with(dframe, pmax(col1, su))
i <- which(dframe[-1] == com_num, arr.ind = TRUE)
dframe$com <- names(dframe[-1])[i[, 2]]
dframe$com_num <- com_num
dframe
# id col1 su com com_num
#1 1 4.2 1.2 col1 4.2
#2 2 1.2 2.4 su 2.4