基于行数设置数据子集True

yc0p9oo0 于 2023-06-27 发布在其他

关注(0)|答案(1)|浏览(109)

在R中，我有一个包含10列的数据集。这些列是二进制的，其中1表示真，2表示假。我需要对数据进行子集化，以找出只有2列等于1，只有3列等于1等的次数问题是--我不在乎哪个3等于1，我只需要3的任何组合等于1的次数的数据。
这是我找到这个的唯一方法，但是我必须为每个组合编写所有这些代码，这对于1023种可能的组合来说太繁琐了。
a1x3 <- filter(ALL_STATES_TBL, ace3 == "1" & ace4 == "2" & ace5 == "2" & ace6 == "2" & ace7 == "2" & ace8 == "2" & ace9 == "2")a1x3t <- filter(a1x3, ace1 == "3")a1x3tt <- filter(a1x3, ace1 == "4")a1x3 <- rbind(a1x3t, a1x3tt)
任何帮助将不胜感激！[另外，代码是这样写的，因为ace 1有4个选项而不是2个]。

来源：https://stackoverflow.com/questions/76560079/subsetting-data-based-on-number-of-rows-true

1条答案

按热度按时间

xytpbqjk1#

我们可以通过n列迭代子集，等于== 1，并计算rowSums是n的次数。combn执行所有组合工作。

f <- \(x) {
  sq <- seq_along(x)[-1]
  sapply(setNames(sq, sq), \(n) 
         sum(unlist(combn(ncol(x), n, \(j) rowSums(x[, j] == 1) == n))))
}

考虑此数据框。

dat1
#   V1 V2 V3 V4
# 1  1  1  2  2
# 2  1  2  2  1
# 3  1  2  1  2

f(dat1)
# 2 3 4 
# 3 0 0

或者这一个

dat2
#   V1 V2 V3 V4 V5 V6 V7 V8 V9 V10
# 1  1  2  1  1  1  2  2  2  1   2
# 2  1  2  2  2  1  1  2  1  1   2
# 3  1  2  1  2  1  1  2  2  2   2
# 4  1  1  2  2  1  1  2  2  2   2
# 5  2  2  1  2  1  1  1  2  2   1

f(dat2)
#  2  3  4  5  6  7  8  9 10 
# 42 38 17  3  0  0  0  0  0

数据：*

set.seed(42)
dat1 <- as.data.frame(matrix(sample(1:2, 12, replace=TRUE), 3, 4))

set.seed(42)
dat2 <- as.data.frame(matrix(sample(1:2, 50, replace=TRUE), 5, 10))

赞(0）回复(0）举报 2023-06-27

我来回答

基于行数设置数据子集True

1条答案

相关问题

热门标签

最新问答