如果元素是同一向量的一部分，如何创建一个值为 1 的二元向量？

Question

我想创建一个由二进制文件组成的所谓匹配向量。除非元素属于同一变量，否则所有数字都应为零。

这是一个示例：

dataset=("a","b","c","d","x","y","z")
var1=c("a","b","y","z")
var2=c("c","d","x")

因此，我有一个数据集，其中第一行中包含所有变量。现在我创建两个组：var1 和 var2。

元素“a”的匹配向量应该如下所示：

matching_a=c(1,1,0,0,0,1,1)

这些数字对应于我的数据集。如果我的数据集中的变量位于同一组中，则我的匹配向量中应该有 1，否则为 0。

然而，我的实际数据集太大，无法手动完成。有谁明白我想做什么吗？

Answer 1

使用 ifelse 函数和 %in% 运算符。

matching_a <-  ifelse(dataset %in% var1, 1, 0)

matching_a
# [1] 1 1 0 0 0 1 1

Answer 2

> output1 = 1 * dataset %in% var1
> output2 = 1 * dataset %in% var2
> output1
[1] 1 1 0 0 0 1 1
> output2
[1] 0 0 1 1 1 0 0

另外，如果您要进行的匹配比 var1 和 var2 多，则将其扩展为以下内容会很有用：

> vars = list(var1, var2)
> 1 * sapply(vars, function(x) dataset %in% x)
     [,1] [,2]
[1,]    1    0
[2,]    1    0
[3,]    0    1
[4,]    0    1
[5,]    0    1
[6,]    1    0
[7,]    1    0

Answer 3

我看到约翰·科尔比已经采取了我要建议的道路，但我认为我会让它更明确。

二元函数%in% 返回一个逻辑向量并乘以 1，强制转换为“数字” 模式。这也可以通过以下方式完成：

matching_a <- as.numeric(dataset %in% x) # Or

matching_a <- 0 + (dataset %in% x)

您还应该查看 %in% 函数所基于的 ?match 。

Answer 4

我使用了上面约翰的方法（和麦克斯的解决方案）的轻微变化来生成“二进制向量”列表（用于多个匹配），如下所示：

library("plyr")

dataset<-c("a","b","c","d","x","y","z")
var1<-c("a","b","y","z")
var2<-c("c","d","x")
vars <- list(var1, var2)

binaryLst <- lapply(vars ,function(x){ifelse(dataset %in% x, 1, 0)})

输出：

> binaryLst
[[1]]
[1] 1 1 0 0 0 1 1

[[2]]
[1] 0 0 1 1 1 0 0