expand.grid с отдельными переменным для каждого столбца

Question

Я хотел бы достичь в R следующего: data.frame

i1 i2 i3 
1 A1 A2 A3 
2 No A2 A3 
3 A1 No A3 
4 No No A3 
5 A1 A2 No 
6 No A2 No 
7 A1 No No 
8 No No No 

В каждой колонке переменные может быть либо Сцепленная строкой «А» и номер столбца или «Нет». data.frame должен содержать все возможные комбинации.

Моя идея заключалась в использовании expand.grid, но я не знаю, как создать list динамически. Или есть лучший подход?

expand.grid(list(c("A1", "No"), c("A2", "No"), c("A3", "No"))) 

Answer 1

Я думаю, вы можете создать свою собственную функцию помощника, что-то вроде этого

MyList <- function(n) expand.grid(lapply(paste0("A", seq_len(n)), c, "No")) 

Тогда просто передать его количество элементов (например, 3)

MyList(3) 
# Var1 Var2 Var3 
# 1 A1 A2 A3 
# 2 No A2 A3 
# 3 A1 No A3 
# 4 No No A3 
# 5 A1 A2 No 
# 6 No A2 No 
# 7 A1 No No 
# 8 No No No 

---

В качестве альтернативы вы также можете попробовать data.table с эквивалентом CJ, который должен быть намного более эффективным, чем expand.grid для большого n

library(data.table) 
DTCJ <- function(n) do.call(CJ, lapply(paste0("A", seq_len(n)), c, "No")) 
DTCJ(3) # will return a sorted cross join 
# V1 V2 V3 
# 1: A1 A2 A3 
# 2: A1 A2 No 
# 3: A1 No A3 
# 4: A1 No No 
# 5: No A2 A3 
# 6: No A2 No 
# 7: No No A3 
# 8: No No No 

Answer 2

Другой вариант заключается в использовании Map с expand.grid

n <- 3 
expand.grid(Map(c, paste0('A', seq_len(n)), 'NO')) 

Или

expand.grid(as.data.frame(rbind(paste0('A', seq_len(n)),'NO'))) 

Answer 3

Другой вариант, используя только самые основные функции в R, является использование индексы:

df <- data.frame(V1 = c('A','A','A', 'A',rep('No',4)), V2 = c('A','A','No','No','A','A','No','No'), V3 = c('A','No','A','No','A','No','A','No'), stringsAsFactors = FALSE) 

, чтобы получить строки и Col индексы элементов, которые мы должны изменить:

rindex <- which(df != 'No') %% nrow(df) 
cindex <- ceiling(which(df != 'No')/nrow(df)) 

решение в основном один вкладыш:

df[matrix(c(rindex,cindex),ncol=2)] <- paste0(df[matrix(c(rindex,cindex),ncol=2)],cindex) 

> df 

    V1 V2 V3 
1 A1 A2 A3 
2 A1 A2 No 
3 A1 No A3 
4 A1 No No 
5 No A2 A3 
6 No A2 No 
7 No No A3 
8 No No No