使用范围作为变量从数据框中选择记录

Question

我有一个数据框“表”，其中有一列名为“ID”。 “ID”的值从 1 到 100。数据框还有一个名为“weight”的列。

我有一个函数“calc_mean”，它有一个变量来选择“ID”：

calc_mean <- function(id=1:100)

当我调用我的函数时，我希望能够获取这个“ID”的一个子集，例如：

calc_mean(30:35)

在我的函数中编写以下代码时，这必须计算我的列“权重”的平均值：

mean(table$weight[,id])

但我收到以下错误：

[1] 不适用 警告信息： 在 mean.default(table$weight[, id]) 中： 参数不是数字或逻辑：返回 NA

怎么了？

或者，如果我可以将此数据框“table”的一个子集制作成另一个名为“table2”的数据框，其中只有我感兴趣的 ID，我会很高兴。然后我会想到以下代码：

for(i in id){
table2 <- table[table$ID == i,]
}

接着是：

mean(table2$weight)

但是，这给了我错误：

[1] 不适用 警告信息： 在 mean.default(table2$weight) 中： 参数不是数字或逻辑：返回 NA

这里有什么问题？

================================================ ===============================

抱歉，由于某些原因，我想先隐藏我的真实代码，但现在将显示真实代码：

pollutantmean <- function(dummy_dir, pollutant, id = 1:332) {
pollutant <- c("sulfate", "nitrate")
directory <- "C:\\Users\\kieken\\Dropbox\\science\\R programming\\specdata"
setwd(directory)
files <- list.files(directory)
data.list <- lapply(files, read.csv)
data.cat <- do.call(rbind, data.list)
good <- complete.cases(data.cat)
data.clean <- data.cat[good,]

data.ID <- subset(data.clean, ID %in% id)
mean(data.ID[,pollutant])

}
pollutantmean("specdata", "nitrate", 70:72)

这段代码给了我以下错误：

[1] 不适用 警告信息： 在 mean.default(data.ID[, 污染物]) 中： 参数不是数字或逻辑：返回 NA

Answer 1

在我的函数中编写以下代码时，这必须计算我的列“权重”的平均值：

mean(table$weight[,id])

这里的逗号没有意义。 table$weight 是一个向量，这意味着它只有一维，而不是二维。因此你应该使用mean(table$weight[id])。

for(i in id){ table2 <- table[table$ID == i,] } 

接着是：

mean(table2$weight)

请注意，每次在 for 函数内循环时，都会将 table2 替换为与 table 不同的行。要创建一个子集，您可以使用

table2 <- table[id,]

或

table2 <- subset(table, ID %in% id)

Answer 2

如果您只想计算“id”子集的平均“权重”，则可以使用with。以下代码计算给定“id”范围内的平均权重

# data example
table <- data.frame(id=1:100, weight=runif(100,60,95))

with(table, mean(weight[id %in% 30:35]))