R语言处理DataFrame：将大于0的值替换为列号的方法求助

需将类群相对丰度的DataFrame转换为存在/缺失格式（用于绘制韦恩图）：将所有大于0的值替换为对应列的列号，0值保留不变。

示例输入数据：

尝试的代码未达到预期效果：

# 最初尝试：整列替换为列号，不符合需求
for(i in 1:ncol(x)) {
  x[ ,i] <- i 
}

# 添加条件判断后仍失败
for(i in 1:ncol(x)) {
  if(x == 0) {
    return(0)
  }
    x[,i] <- i
}

方法1：修正循环逻辑 #

原循环未针对列内单个元素做条件判断，正确写法如下：

# 假设数据框名为x
for (i in 1:ncol(x)) {
  # 定位当前列中大于0的元素，替换为列号i
  x[x[, i] > 0, i] <- i
}

方法2：base R向量化实现（更高效） #

用sapply避免循环，代码更简洁：

# 处理数据
x_processed <- sapply(1:ncol(x), function(i) {
  ifelse(x[, i] > 0, i, 0)
})
# 转换回数据框并恢复列名
x_processed <- as.data.frame(x_processed)
colnames(x_processed) <- colnames(x)

方法3：tidyverse/dplyr实现 #

若使用tidyverse工具链，可通过mutate_all完成：

library(dplyr)

x_processed <- x %>%
  mutate_all(.funs = function(col_val, idx) {
    ifelse(col_val > 0, idx, 0)
  }, idx = 1:ncol(x))

转换后示例数据结果：

内容的提问来源于stack exchange，提问作者A_ga_ga_ghost