Question

我编写了一个运行数据帧列的代码，如果其中任何一个数组中的数字为1，则返回TRUE，并将值发送到与数据帧的列大小相同的向量。我想知道是否有办法简化下面的代码片段，因为我将不得不重复几个数字。

n1 <- (tab[, 2]==1|  tab[, 3]==1 | tab[, 4]==1 | tab[, 5]==1 | 
     tab[, 6]==1 | tab[, 7]==1 | tab[, 8]==1 | tab[, 9]==1 |
     tab[, 10]==1 | tab[, 11]==1 | tab[, 12]==1 | tab[, 13]==1 |
     tab[, 14]==1 | tab[, 15]==1 | tab[, 16]==1)

Answer 1

一种可能的解决方案如下：在数据框中搜索== 1个数字，然后使用import os os.environ['TF_CPP_MIN_LOG_LEVEL']='2' import tensorflow as tf运算符减少其行数：

对于此示例，它将为您提供以下输出：

tab <- data.frame(a = 1:10, b = 2:11)

apply(tab == 1, 1, function(x) {
  Reduce("|", x)
})

或者更简单的解决方案是：

[1]  TRUE FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE FALSE

Answer 2

其他评论和答案可以奏效，但我建议他们在处理data.frame时鼓励不良行为。首先，apply和rowSums期望matrix作为数据，并且如果给出data.frame，则会很乐意强制执行此操作。如果data.frame列中的任何一列为character，则所有列都将转换为character。某些操作可能仍然按预期工作（例如== 1，因为它实际上会== "1" ...虽然某些舍入错误可能会导致不良影响），但任何数学操作都无效。

举个例子，

n <- 20
set.seed(2)
tab <- data.frame(
  a = as.character(sample(n, replace = FALSE)),
  b1 = sample(5, size = n, replace = TRUE),
  b2 = sample(5, size = n, replace = TRUE),
  stringsAsFactors = FALSE
)
str(tab)
# 'data.frame': 20 obs. of  3 variables:
#  $ a : chr  "4" "14" "11" "3" ...
#  $ b1: int  4 2 5 1 2 3 1 2 5 1 ...
#  $ b2: int  5 2 1 1 5 4 5 2 3 5 ...

apply(tab, 1, function(y) any(y == 1))
#  [1] FALSE FALSE  TRUE  TRUE FALSE FALSE  TRUE FALSE FALSE  TRUE  TRUE  TRUE FALSE FALSE FALSE FALSE FALSE  TRUE FALSE  TRUE
apply(tab, 1, sum)
# Error in FUN(newX[, i], ...) : invalid 'type' (character) of argument

rowSums(tab == 1)
#  [1] 0 0 1 2 0 0 1 0 0 1 2 2 0 0 0 0 0 1 0 1
rowSums(tab)
# Error in rowSums(tab) : 'x' must be numeric

有一些简单的方法可以解决这个问题。根据您的示例，列2:16似乎是数字和您关注的列。如果是这种情况，那么您可以安全地使用以下任何一个：

rowSums(tab[,2:16] == 1)                      # Frank's comment
apply(tab[,2:16], 1, function(y) any(y == 1)) # suggested by You-leee's answer

（前者相当具体，后者可以扩展到其他功能）。如果只有一个非numeric，则一次可以

rowSums(tab[,-1,drop=FALSE] == 1)
apply(tab[,-1,drop=FALSE], 1, function(y) any(y == 1))

第三种技术是在运行时确定选择哪些列：

isnum <- sapply(tab, is.numeric)
Reduce(`|`, lapply(tab[isnum], function(y) any(y == 1)))

这有点复杂，因为lapply的回报是list，但它仍然正常。要意识到isnum的使用也可以基于列名，使用grepl之类的东西。这种方法也非常健壮，因为如果没有列匹配，它就不会出错。

简化R中的代码

2 个答案: