遍历数据帧并根据条件[R]更改值

时间:2019-09-18 17:11:24

标签: r loops if-statement

必须要记帐,因为这个for循环序列已经困扰了我很长时间了。

我在R中有一个数据帧,具有1000行和10列,每个值的范围为1:3。我想重新编码每个条目,以便:1 == 3,2 == 2,3 == 1。我知道可以使用更简单的方法,例如对每个列进行子设置并对条件进行硬编码,但这并不总是理想的,因为我使用的许多数据集最多有100列。

我想使用嵌套循环来完成此任务-到目前为止,这就是我的目的:

for(i in 1:nrow(dat_trans)){
  for(j in length(dat_trans)){
    if(dat_trans[i,j] == 1){
      dat_trans[i,j] <- 3
    } else if(dat_trans[i,j] == 2){
      dat_trans[i,j] <- 2
    } else{
      dat_trans[i,j] <- 1
    }
  }
}

因此,我遍历第一列,获取每个值并根据if / else的条件对其进行更改,我仍在学习R,因此,如果我的代码中有任何指针,请随时指出。

编辑:代码

4 个答案:

答案 0 :(得分:2)

R是向量化语言,因此您实际上不需要内部循环。
另外,如果您注意到4-“旧值” =“新值”,则可以消除if语句。

for(i in 1:ncol(dat_trans)){
        dat_trans[,i] <- 4-dat_trans[,i]
}

外部循环现在仅在列中进行10次迭代,而不是在所有行中进行1000次迭代。这将大大提高性能。

答案 1 :(得分:1)

这种类型的操作是交换操作。不使用for循环交换值的方法很多。

要设置一个简单的数据框:

df <- data.frame(
  col1 = c(1,2,3),
  col2 = c(2,3,1),
  col3 = c(3,1,2)
)

使用虚拟值:

df[df==1] <- 4
df[df==3] <- 1
df[df==4] <- 3

使用临时变量:

dftemp <- df
df[dftemp==1] <- 3
df[dftemp==3] <- 1

使用乘法/除法和加法/减法:

df <- 4 - df

使用布尔运算:

df <- (df==1) * 3 + (df==2) * 2 + (df==3) * 1

使用按位异或(如果您确实需要速度):

df[df!=2] <- sapply(df, function(x){bitwXor(2,x)})[df!=2]

如果需要嵌套的for循环,则切换功能是个不错的选择。

for(i in seq(ncol(df))){
  for(j in seq(nrow(df))){
    df[j,i] <- switch(df[j,i],3,2,1)
  }
}

如果值的索引值不如1、2和3,则可以使用文本。

for(i in seq(ncol(df))){
  for(j in seq(nrow(df))){
    df[j,i] <- switch(as.character(df[j,i]),
                      "1" = 3,
                      "2" = 2,
                      "3" = 1)
  }
}

答案 2 :(得分:0)

这听起来像是merge / join操作。

set.seed(42)
dat_trans <- as.data.frame(
  setNames(lapply(1:3, function(ign) sample(1:3, size=10, replace=TRUE)),
           c("V1", "V2", "V3"))
)
dat_trans
#    V1 V2 V3
# 1   3  2  3
# 2   3  3  1
# 3   1  3  3
# 4   3  1  3
# 5   2  2  1
# 6   2  3  2
# 7   3  3  2
# 8   1  1  3
# 9   2  2  2
# 10  3  2  3

newvals <- data.frame(old = c(1, 3), new = c(3, 1))
newvals
#   old new
# 1   1   3
# 2   3   1

使用dplyrtidyr

library(dplyr)
library(tidyr) # gather, spread
dat_trans %>%
  mutate(rn = row_number()) %>%
  gather(k, v, -rn) %>%
  left_join(newvals, by = c("v" = "old")) %>%
  mutate(v = if_else(is.na(new), v, new)) %>%
  select(-new) %>%
  spread(k, v) %>%
  select(-rn)
#    V1 V2 V3
# 1   1  2  1
# 2   1  1  3
# 3   3  1  1
# 4   1  3  1
# 5   2  2  3
# 6   2  1  2
# 7   1  1  2
# 8   3  3  1
# 9   2  2  2
# 10  1  2  1

(之所以需要rn,可能是由于我使用的是tidyr的旧版本:我是0.8.2,尽管最近发布了1.0.0。在spread / gather上进行了大量的增强/工作,并引入了pivot_*函数,这可能会更加平滑。如果您使用的是最新版本,请尝试使用{{1} }部分。)


或者使用“重新编码”心态的更直接的方法:

rn

答案 3 :(得分:0)

您可以使用分配矩阵ammatch()的{​​{1}}属性的每个值都位于df1的第1列,但选择第2列,然后将其分配给am。当然是df1

lapply()

数据

df1
#   V1 V2 V3
# 1  1  2  1
# 2  1  2  1
# 3  1  1  2
# 4  1  3  2
# 5  2  3  2

am <- matrix(c(1, 2, 3, 3, 2, 1), 3)
am
#      [,1] [,2]
# [1,]    1    3
# [2,]    2    2
# [3,]    3    1

df1[] <- lapply(df1, function(x) am[match(x, am[,1]), 2])
df1
#   V1 V2 V3
# 1  3  2  3
# 2  3  2  3
# 3  3  3  2
# 4  3  1  2
# 5  2  1  2