如何将 dataframe 中的某些值替换为 r 中的列名？

Question

我正在尝试用变量名替换 dataframe 中的某些字符值。 例如，将“1”的所有值替换为该值对应的变量名。

所以我想采取这样的 dataframe ：

最后得到一个 dataframe 像这样：

我编写了一个有效的 for 循环（请参见下文），但是当我尝试将 for 循环编写为 function 时，dataframe 会保留原始值并且我没有收到任何错误消息。 感谢您对调试 function 的任何帮助。 这是一个可重现的示例。

# Create Dataframes ------------------------------------------------------------

testData1 <- data.frame(x = c('1', '0', '0', '0', '1'),
                        y = c('1', '1', '0', '1', '0'),
                        z = c('0', '0', '1', '0', '0'))

testData2 <- data.frame(x = c('1', '0', '0', '0', '1'),
                        y = c('1', '1', '0', '1', '0'),
                        z = c('0', '0', '1', '0', '0'))

# For Loop ---------------------------------------------------------------------

for (i in 1:nrow(testData1)) {
  for (j in 1:ncol(testData1)) {
    
    if (testData1[i, j] == '1') {
      
      testData1[i, j] <- colnames(testData1)[j]
      
    }
  }
}

# Function ---------------------------------------------------------------------

# takes a dataframe where all vectors are characters and replaces the value of 
# '1' with the name of the vector

replaceValues <- function(data) {
  
  for (i in 1:nrow(data)) {
    for (j in 1:ncol(data)) {
      
      if (data[i, j] == '1') {
        
        data[i, j] <- colnames(data)[j]

      }
    }
  }
}

replaceValues(testData2)

Answer 1

我们可以做向量化操作

i1 <- testData1 == 1
testData1[i1] <- names(testData1)[col(testData1)[i1]]

-输出

> testData1
  x y z
1 x y 0
2 0 y 0
3 0 0 z
4 0 y 0
5 x 0 0

Answer 2

通过@akrun 和@Dominik Żabiński

# Create Dataframes ------------------------------------------------------------

testData2 <- data.frame(x = c('1', '0', '0', '0', '1'),
                        y = c('1', '1', '0', '1', '0'),
                        z = c('0', '0', '1', '0', '0'))

# Function ---------------------------------------------------------------------

# takes a dataframe where all vectors are characters and replaces the value of 
# '1' with the name of the vector

replaceValues <- function(data) {
  
  for (i in 1:nrow(data)) {
    for (j in 1:ncol(data)) {
      
      if (data[i, j] == '1') {
        
        data[i, j] <- colnames(data)[j]

      }
    }
  }
  return(data)
}

testData2 <- replaceValues(testData2)