Question

我正在尝试制作矩阵的子集。每个子集取决于一组逻辑条件。例如，第一个列表的第一个子集（R[[1]][[1]]）对应于矩阵的最后一列等于1或3等。我想在循环中使用它。

这是我的数据示例：

l = list(
  list(c(1, 3),c(2, 3)),
  list(c(1,5),c(2,5,6,7),c(3,6,8),c(4,7,8))
)

a1 <- array(1, dim=c(12,10))
a2 <- array(1, dim=c(18,10))
v1 <- c(1,1,3,1,2,3,1,1,3,1,2,2)
v2 <- c(1,1,3,1,5,2,4,5,5,5,7,8,1,4,5,8,2,4)<br/>
m1 <- as.matrix(cbind(a1,v1))
m2 <- as.matrix(cbind(a2,v2))
m <- list(m1,m2)

这是我使用的代码：

R <- rep(list(list()),length(m))
  for(i in 1:length(m)) {
    for(j in 1:length(l[[i]])) {
      R[[i]][[j]] <- m[[i]][m[[i]][,dim(m[[i]])[2]] == l[[i]][[j]],]
}}

我得到以下结果，这不是我想要的：

[[1]]
[[1]][[1]]
[1,] 1 1 1 1 1 1 1 1 1 1  1
[2,] 1 1 1 1 1 1 1 1 1 1  3
[3,] 1 1 1 1 1 1 1 1 1 1  1
[[1]][[2]]
[1,] 1 1 1 1 1 1 1 1 1 1  2
[2,] 1 1 1 1 1 1 1 1 1 1  3
[3,] 1 1 1 1 1 1 1 1 1 1  2
[[2]]
[[2]][[1]]
[1,] 1 1 1 1 1 1 1 1 1 1  1
[2,] 1 1 1 1 1 1 1 1 1 1  5
[3,] 1 1 1 1 1 1 1 1 1 1  5
[4,] 1 1 1 1 1 1 1 1 1 1  1
[[2]][[2]]
[1,] 1 1 1 1 1 1 1 1 1 1  5
[2,] 1 1 1 1 1 1 1 1 1 1  2
[[2]][[3]]
 1  1  1  1  1  1  1  1  1  1  8 
[[2]][[4]]
[1,] 1 1 1 1 1 1 1 1 1 1  4
[2,] 1 1 1 1 1 1 1 1 1 1  7
[3,] 1 1 1 1 1 1 1 1 1 1  8

这是我想要得到的结果：

[[1]]
[[1]][[1]]
 [1,] 1 1 1 1 1 1 1 1 1 1  1
 [2,] 1 1 1 1 1 1 1 1 1 1  1
 [3,] 1 1 1 1 1 1 1 1 1 1  3
 [4,] 1 1 1 1 1 1 1 1 1 1  1
 [5,] 1 1 1 1 1 1 1 1 1 1  3
 [6,] 1 1 1 1 1 1 1 1 1 1  1
 [7,] 1 1 1 1 1 1 1 1 1 1  1
 [8,] 1 1 1 1 1 1 1 1 1 1  3
 [9,] 1 1 1 1 1 1 1 1 1 1  1
[[1]][[2]]
[1,] 1 1 1 1 1 1 1 1 1 1  3
[2,] 1 1 1 1 1 1 1 1 1 1  2
[3,] 1 1 1 1 1 1 1 1 1 1  3
[4,] 1 1 1 1 1 1 1 1 1 1  3
[5,] 1 1 1 1 1 1 1 1 1 1  2
[6,] 1 1 1 1 1 1 1 1 1 1  2
[[2]]
[[2]][[1]]
 [1,] 1 1 1 1 1 1 1 1 1 1  1
 [2,] 1 1 1 1 1 1 1 1 1 1  1
 [3,] 1 1 1 1 1 1 1 1 1 1  1
 [4,] 1 1 1 1 1 1 1 1 1 1  5
 [5,] 1 1 1 1 1 1 1 1 1 1  5
 [6,] 1 1 1 1 1 1 1 1 1 1  5
 [7,] 1 1 1 1 1 1 1 1 1 1  5
 [8,] 1 1 1 1 1 1 1 1 1 1  1
 [9,] 1 1 1 1 1 1 1 1 1 1  5
[[2]][[2]]
[1,] 1 1 1 1 1 1 1 1 1 1  5
[2,] 1 1 1 1 1 1 1 1 1 1  2
[3,] 1 1 1 1 1 1 1 1 1 1  5
[4,] 1 1 1 1 1 1 1 1 1 1  5
[5,] 1 1 1 1 1 1 1 1 1 1  5
[6,] 1 1 1 1 1 1 1 1 1 1  7
[7,] 1 1 1 1 1 1 1 1 1 1  5
[8,] 1 1 1 1 1 1 1 1 1 1  2
[[2]][[3]]
[1,] 1 1 1 1 1 1 1 1 1 1  3
[2,] 1 1 1 1 1 1 1 1 1 1  8
[3,] 1 1 1 1 1 1 1 1 1 1  8
[[2]][[4]]
[1,] 1 1 1 1 1 1 1 1 1 1  4
[2,] 1 1 1 1 1 1 1 1 1 1  7
[3,] 1 1 1 1 1 1 1 1 1 1  8
[4,] 1 1 1 1 1 1 1 1 1 1  4
[5,] 1 1 1 1 1 1 1 1 1 1  8
[6,] 1 1 1 1 1 1 1 1 1 1  4

Answer 1

你在找这个吗？

如果您希望同样应用列表m的所有元素：

k<-lapply(m,function(x) x[x[,11] %in% c(1,3),])

> k
[[1]]
                          v1
 [1,] 1 1 1 1 1 1 1 1 1 1  1
 [2,] 1 1 1 1 1 1 1 1 1 1  1
 [3,] 1 1 1 1 1 1 1 1 1 1  3
 [4,] 1 1 1 1 1 1 1 1 1 1  1
 [5,] 1 1 1 1 1 1 1 1 1 1  3
 [6,] 1 1 1 1 1 1 1 1 1 1  1
 [7,] 1 1 1 1 1 1 1 1 1 1  1
 [8,] 1 1 1 1 1 1 1 1 1 1  3
 [9,] 1 1 1 1 1 1 1 1 1 1  1

[[2]]
                         v2
[1,] 1 1 1 1 1 1 1 1 1 1  1
[2,] 1 1 1 1 1 1 1 1 1 1  1
[3,] 1 1 1 1 1 1 1 1 1 1  3
[4,] 1 1 1 1 1 1 1 1 1 1  1
[5,] 1 1 1 1 1 1 1 1 1 1  1

Answer 2

这是一个适用于%in%的解决方案：

R <- rep(list(list()),length(m)) <br/>
for(i in 1:length(m)) { <br/>
for(j in 1:length(l[[i]])) { <br/>
R[[i]][[j]] <- m[[i]][m[[i]][,dim(m[[i]])[2]] %in% l[[i]][[j]],] <br/>
}}

使用逻辑运算符或（|）子集for循环中的矩阵

2 个答案: