Использование цикла для выбора столбцов с использованием разных векторов
Допустим, у меня есть 3 вектора (10 строк):
X <- c(1,1,0,1,0, 1,1, 0, NA,NA)
H <- c(0,0,1,0,NA,1,NA,1, 1, 1 )
I <- c(0,0,0,0,0, 1,NA,NA,NA,1 )
data.frame Y содержит 10 столбцов и 6 строк:
1 2 3 4 5 6 7 8 9 10
0 1 0 0 1 1 1 0 1 0
1 1 1 0 1 0 1 0 0 0
0 0 0 0 1 0 0 1 0 1
1 0 1 1 0 1 1 1 0 0
0 0 0 0 0 0 1 0 0 0
1 1 0 1 0 0 0 0 1 1
Я хотел бы использовать вектор X, H en I, чтобы сделать выбор столбцов в data.frame Y, используя "1" и "0" в векторе в качестве критерия выбора. Таким образом, результаты для вектора X с использованием "1" в качестве критерия выбора должны быть:
X <- c(1,1,0,1,0, 1,1, 0, NA,NA)
1 2 4 6 7
0 1 0 1 1
1 1 0 0 1
0 0 0 0 0
1 0 1 1 1
0 0 0 0 1
1 1 1 0 0
Для вектора H, используя "1" в качестве критерия выбора:
H <- c(0,0,1,0,NA,1,NA,1, 1, 1 )
3 6 8 9 10
0 1 0 1 0
1 0 0 0 0
0 0 1 0 1
1 1 1 0 0
0 0 0 0 0
0 0 0 1 1
Для вектора я использовал "1" в качестве критерия выбора:
I <- c(0,0,0,0,0, 1,NA,NA,NA,1 )
6 10
1 0
0 0
0 1
1 0
0 0
0 1
Для удобства и скорости я бы хотел использовать петлю. Это может быть что-то вроде этого:
all.ones <- lapply[,function(x) x %in% 1]
В результате (all.ones) результат для каждого вектора должен оставаться отдельным. Например:
X 1,2,4,6,7
H 3,6,8,9,10
I 6,10
3 ответа
Стандартный способ сделать это с помощью %in% оператор:
Y[, X %in% 1]
Чтобы сделать это для нескольких векторов (при условии, что вы хотите операцию AND):
mylist = list(X, H, I, D, E, K)
Y[, Reduce(`&`, lapply(mylist, function(x) x %in% 1))]
Проблема в использовании, использовании which чтобы обойти это. Учтите следующее:
x <- c(1,0,1,NA)
x[x==1]
[1] 1 1 NA
x[which(x==1)]
[1] 1 1
Как насчет этого?
idx <- which(X==1)
Y[,idx]
РЕДАКТИРОВАТЬ: для шести векторов, сделать
idx <- which(X==1 & H==1 & I==1 & D==1 & E==1 & K==1)
Y[,idx]
замещать & с | если вы хотите все столбцы Y, где хотя бы один из списков имеет 1.