R-用非零值的列均值替换多个列

Question

我有类似的数据：

aye <- c(0,0,3,4,5,6)
bee <- c(3,4,0,0,7,8)
see <- c(9,8,3,5,0,0)
df <- data.frame(aye, bee, see)

我正在寻找一种基于数据框中每个列的平均值创建列的简洁方法，其中零保持为零。

获得零的平均值：

df2 <- as.data.frame(t(apply(df, 2, function(x) mean(x[x>0]))))

我无法弄清楚如何简单地用均值零替换列中的值。到目前为止，我的方法是：

df$aye <- ifelse(df$aye == 0, 0, df2$aye)
df$bee <- ifelse(df$bee == 0, 0, df2$bee)
df$see <- ifelse(df$see == 0, 0, df2$see)

，但这与许多变量变得混乱 - 将其包装在一个函数中会很不错。

感谢您的帮助！

Answer 1

为什么我们不能只使用

data.frame(lapply(dat, function (u) ave(u, u > 0, FUN = mean)))
#  aye bee  see
#1 0.0 5.5 6.25
#2 0.0 5.5 6.25
#3 4.5 0.0 6.25
#4 4.5 0.0 6.25
#5 4.5 5.5 0.00
#6 4.5 5.5 0.00

注意，我使用dat而不是df作为数据框的名称。df是R中的功能，不要掩盖它。

Answer 2

我们可以将应用函数作为数字向量的结果保持在 x中。

x <- apply(df, 2, function(x){ mean(x[x>0])})
df[which(df!=0, arr.ind = T)] <- x[ceiling(which(df!=0)/nrow(df))]
df
#  aye bee  see
#1 0.0 5.5 6.25
#2 0.0 5.5 6.25
#3 4.5 0.0 6.25
#4 4.5 0.0 6.25
#5 4.5 5.5 0.00
#6 4.5 5.5 0.00

进一步分解代码以解释工作

给出值不为零

的索引

which(df! = 0)
#[1]  3  4  5  6  7  8 11 12 13 14 15 16

此行决定我们将要从x

选择哪些索引

ceiling(which(df!=0)/nrow(df))
#[1] 1 1 1 1 2 2 2 2 3 3 3 3
x[ceiling(which(df!=0)/nrow(df))]
#aye  aye  aye  aye  bee  bee  bee  bee  see  see  see  see 
#4.50 4.50 4.50 4.50 5.50 5.50 5.50 5.50 6.25 6.25 6.25 6.25 

现在替换上述值，其中值不等于dataFrame

中的值。

df[which(df!=0, arr.ind = T)] <- x[ceiling(which(df!=0)/nrow(df))]

Answer 3

尝试将您已经拥有的内容重新安排到zeroless_mean函数中，然后在数据的每一列上使用apply。Frame：

# Data
aye <- c(0,0,3,4,5,6)
bee <- c(3,4,0,0,7,8)
see <- c(9,8,3,5,0,0)
dff <- data.frame(aye, bee, see)
# Function
zeroless_mean <- function(x) ifelse(x==0,0,mean(x[x!=0]))
# apply
data.frame(apply(dff, 2, zeroless_mean))
# Output
  aye bee  see
1 0.0 5.5 6.25
2 0.0 5.5 6.25
3 4.5 0.0 6.25
4 4.5 0.0 6.25
5 4.5 5.5 0.00
6 4.5 5.5 0.00

我希望这会有所帮助。