Network Canvas导出数据转因子列全为NA问题求助

问题背景 #

从Network Canvas（NC）导出的数据中，分类变量以多列布尔值形式存在（每个类别对应一列），需要重编码为单列因子变量（如性别、种族）。使用NC官网提供的catToFactor函数后，生成的因子列结构正确，但所有记录值均为NA，无有效数据。

原代码 #

catToFactor <- function(dataframe,variableName) {
    fullVariableName <- paste0(variableName,"_")
    catVariables <- grep(fullVariableName, names(dataframe), value=TRUE)
    # Check if variable exists
    if (identical(catVariables, character(0))){
      stop(paste0("Cannot find variable named -",variableName,"- in the data"))
    # Check if "true" in multiple columns of a single row
    } else if (sum(apply(dataframe[,catVariables], 1, function(x) sum(x %in% "true")>1))>0) {
      stop(paste0("Your variable -",variableName,"  - appears to take multiple values.")) }
    catValues <- sub(paste0('.*',fullVariableName), '', catVariables)
    factorVariable <- c()
    for(i in 1:length(catVariables)){
      factorVariable[dataframe[catVariables[i]]=="true"] <- catValues[i]
    }
    return(factor(factorVariable,levels=catValues))
}

# List of categorical variables in our protocol to convert into factors
categoricalVariablesList <- list('Gender','Race','SexOrient')

# Iterate the list and call our catToFactor function, assigning the result to a new column in our dataframe
for (variable in categoricalVariablesList) {
  alterData[variable] <- catToFactor(alterData, variable)
}

问题原因 #

1. 核心问题：NC导出的布尔列是逻辑型（TRUE/FALSE），而非字符串型的"true"，原代码中用x %in% "true"和dataframe[catVariables[i]]=="true"进行判断，无法匹配逻辑值，导致赋值失败，最终全为NA。
2. 次要问题：factorVariable初始化为空向量，当索引赋值时可能出现长度不匹配的隐性问题。

修复后的代码 #

catToFactor <- function(dataframe, variableName) {
    fullVariableName <- paste0(variableName, "_")
    catVariables <- grep(fullVariableName, names(dataframe), value = TRUE)
    
    # 检查变量是否存在
    if (identical(catVariables, character(0))) {
      stop(paste0("Cannot find variable named -", variableName, "- in the data"))
    }
    
    # 检查是否有行同时存在多个TRUE值（逻辑型判断）
    multi_val_rows <- sum(apply(dataframe[, catVariables], 1, function(x) sum(x) > 1))
    if (multi_val_rows > 0) {
      stop(paste0("Your variable -", variableName, "- appears to take multiple values in ", multi_val_rows, " rows."))
    }
    
    catValues <- sub(paste0('.*', fullVariableName), '', catVariables)
    # 初始化与数据行数一致的NA向量，避免索引问题
    factorVariable <- rep(NA_character_, nrow(dataframe))
    
    for (i in seq_along(catVariables)) {
      # 匹配逻辑型TRUE值
      factorVariable[dataframe[[catVariables[i]]] == TRUE] <- catValues[i]
    }
    
    return(factor(factorVariable, levels = catValues))
}

# 待转换的分类变量列表
categoricalVariablesList <- c('Gender', 'Race', 'SexOrient')

# 批量转换并添加到数据框
for (variable in categoricalVariablesList) {
  alterData[[variable]] <- catToFactor(alterData, variable)
}

关键修改说明 #

• 将所有字符串"true"的判断改为逻辑值TRUE，适配NC导出的列类型
• 初始化factorVariable为与数据行数一致的NA字符向量，避免索引越界或长度不匹配
• 优化多值行的检查逻辑，直接对逻辑值求和，更高效
• 用[[ ]]提取列，比[ ]更稳定（避免返回数据框而非向量）

验证方法：转换后可以用table(alterData$Gender, useNA = "always")查看因子分布，确认NA占比是否符合预期

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