R中的循环条件

问题描述 投票:0回答:1

我有一个相对简单的问题,我无法应用我在互联网上找到的解决方案。假设我们有:

set.seed(20)

data <- data.frame(month = rep(month.name, 25), 
a = rnorm(300, 0, 1), b = runif(300, 0, 7.2))

我想计算一个循环f-test,用于每月每月的a和b列之间的差异。这是我通过使用:

# create some empty vectors to fill in later
pval <- as.double()
ftest <- as.double()
month <- as.character()

# looping through the months

for (i in unique(data$month)){
  print(i)
  # sh.1 <- shapiro.test(data$a[data$month==i])
  # sh.1[2] > 0.05 # apply log if it's smaller than 0.05
  # sh.2 <- shapiro.test(data$b[data$month==i])
  # sh.2[2] > 0.05 # apply log if it's smaller than 0.05
  var.t <- var.test(data$a[data$month==i], data$b[data$month==i])
  f <- round(var.t[[1]],2)
  p <- round(var.t$p.value,2)
  ftest <- append(ftest, f)
  pval <- append(pval, p)
  month <- append(month, i)
}

但是,据我所知,f-test对正态分布非常敏感。因此,我打算使用一个条件进入循环,如果shapiro测试的p值小于0.05,则需要对数据进行对数转换;然后它将用于f-test。

通常情况下,我会以一个ifelse的条件,但我不太确定如何在这里使用它。请帮忙吗?

r loops if-statement normal-distribution
1个回答
1
投票

我相信下面的代码可以满足您的需求。它使用*apply循环,而不是for循环,以使代码更具可读性(我认为)。

首先,我将重新创建数据并确保列a都是正数。

set.seed(20)

data <- data.frame(month = rep(month.name, 25), 
                   a = rnorm(300, 0, 1), b = runif(300, 0, 7.2))

data$a <- abs(data$a)

现在,我不是循环遍历month的唯一值,而是将data.frame拆分为该变量。像这样,结果列表sp中的每个df已经是每个月所有行的df。

sp <- split(data, data$month)
sp <- sp[order(order(month.name))]

正是在这里,如有必要,数据将被转换为log

sp <- lapply(sp, function(DF){
  if(shapiro.test(DF[["a"]])$p.value < 0.05) DF[["a"]] <- log(DF[["a"]])
  if(shapiro.test(DF[["b"]])$p.value < 0.05) DF[["b"]] <- log(DF[["b"]])
  DF
})

并且lapply你想要的测试,var.test,所有这些data.frames。

vartest_list <- lapply(sp, function(DF){
  var.t <- var.test(DF[["a"]], DF[["b"]])
  list(f = var.t[[1]], 
       p.value = var.t$p.value, 
       month = as.character(DF[["month"]][1]))
})

最后,将提取函数[[应用于测试结果是一件简单的事情。这是有效的,因为假设测试函数在类"htest"的R返回对象中除了列表之外别无其他。最后一个提取循环被注释掉了。

ftest <- sapply(vartest_list, '[[', 'f')
pval <- sapply(vartest_list, '[[', 'p.value')
#month <- sapply(vartest_list, '[[', 'month')
© www.soinside.com 2019 - 2024. All rights reserved.