Question

我想在R中创建一个函数来比较“当前行值”和“先前行值”。一种。如果当前行值小于上一行值，则返回单词LOW b。如果当前行值高于上一个值，则返回单词HIGH C。如果当前行值与上一行值相同，则返回单词EQUAL

我想创建一个名为Decision的新列，该词的存储位置如下所示

价值决策

1不适用

2高

1.5低

1.4低

1.4相等

Answer 1

我们可以使用case_when将“值”与上一个“值”（lag）进行比较，并根据其值是较低还是较高来创建“决策”列

library(dplyr)
df1 %>%
 mutate(Decision = case_when(Value < lag(Value) ~ "LOW", 
     Value > lag(Value) ~ "HIGH", Value == lag(Value) ~ "EQUAL"))
#  Value Decision
#1   1.0     <NA>
#2   2.0     HIGH
#3   1.5      LOW
#4   1.4      LOW
#5   1.4    EQUAL

或者另一种选择是采取差异，然后更改label中的factor

with(df1, c(NA_character_, 
    as.character(factor(sign(Value[-length(Value)] - Value[-1]),         
       levels = c(-1, 1, 0), labels = c("HIGH", "LOW", "EQUAL")))))
#[1] NA      "HIGH"  "LOW"   "LOW"   "EQUAL"

或使用for循环

df1$Decision <- rep(NA_character_, nrow(df1))
for(i in 2: nrow(df1)){
    if(df1$Value[i] > df1$Value[i-1]){
       df1$Decision[i] <- "HIGH"
      } else if(df1$Value[i] < df1$Value[i-1]) {

        df1$Decision[i] <- "LOW"
       } else df1$Decision[i] <- "EQUAL"


   }

如果需要作为函数

f1 <- function(dat, colNm, newCol) {
  dat[[newCol]] <- rep(NA_character_, nrow(dat))
    for(i in 2: nrow(dat)){
        if(dat[[colNm]][i] > dat[[colNm]][i-1]){
           dat[[newCol]][i] <- "HIGH"
          } else if(dat[[colNm]][i] < dat[[colNm]][i-1]) {

            dat[[newCol]][i] <- "LOW"
           } else dat[[newCol]][i] <- "EQUAL"


       }
     dat
   }

f1(df1, "Value", "Decision")

数据

df1 <- structure(list(Value = c(1, 2, 1.5, 1.4, 1.4)), class = "data.frame",
     row.names = c(NA, 
-5L))

在R中创建一个函数以比较行

1 个答案:

数据