длинный цикл if else и перекодирование в R - PullRequest
4 голосов
/ 09 февраля 2012

Я знаю, что моя проблема проста, но не для меня.Вот небольшой набор данных.

mark1 <- c("AB", "BB", "AB", "BB", "BB", "AB", "--", "BB")
mark2 <- c("AB", "AB", "AA", "BB", "BB", "AA", "--", "BB")
mark3 <- c("BB", "AB", "AA", "BB", "BB", "AA", "--", "BB")
mark4 <- c("AA", "AB", "AA", "BB", "BB", "AA", "--", "BB")
mark5 <- c("AB", "AB", "AA", "BB", "BB", "AA", "--", "BB")
mark6 <- c("--", "BB", "AA", "BB", "BB", "AA", "--", "BB")
mark7 <- c("AB", "--", "AA", "BB", "BB", "AA", "--", "BB")
mark8 <- c("BB", "AA", "AA", "BB", "BB", "AA", "--", "BB")
mymark <- data.frame (mark1, mark2, mark3, mark4, mark5, mark6, mark7, mark8)
tmymark <- data.frame (t(mymark))
names (tmymark) <- c("P1", "P2","I1", "I2", "I3", "I4", "KL", "MN")

Таким образом, набор данных становится:

      P1 P2 I1 I2 I3 I4 KL MN
mark1 AB BB AB BB BB AB -- BB
mark2 AB AB AA BB BB AA -- BB
mark3 BB AB AA BB BB AA -- BB
mark4 AA AB AA BB BB AA -- BB
mark5 AB AB AA BB BB AA -- BB
mark6 -- BB AA BB BB AA -- BB
mark7 AB -- AA BB BB AA -- BB
mark8 BB AA AA BB BB AA -- BB

Я хочу классифицировать mark1: 8 на основе сравнения P1 и P2 и предоставить код, который создаст новую переменную:

loctype <- NULL

if (tmymark$P1 == "AB" &  tmymark$P2 == "AB"){
       loctype = "<hkxhk>"
       } else {
if (tmymark$P1== "AB" & tmymark$P2 == "BB") {
      loctype = "<lmxll>"
      } else {
      if (tmymark$P1 == "AA" & tmymark$P2 == "AB") {
       loctype = "<nnxnp>"
       } else {
        if (tmymark$P1 == "AA" & tmymark$P2 == "BB") {
        loctype = "MN"
        } else {
        if (tmymark$P1 == "BB" & tmymark$P2 == "AA"){
         loctype = "MN"
         } else {
         if (tmymark$P1 == "--" & tmymark$P2 == "AA"){
         loctype = "NR"
         }  else {
if (tmymark$P1 == "AA" & tmymark$P2 == "--"){
          loctype = "NR"
} else {
    cat ("error wrong input in P1 or P2")
    }} }}}}}

Вот что я пытаюсь сделать, это сравнить значения P1 и P2 и сгенерировать новую переменную.Например, если tmymark $ P1 == "AB" и tmymark $ P2 == "AB", то тип логики должен быть "".Если нет, то вторым условием будет применение и так далее.

Вот мое сообщение об ошибке.

Warning messages:
1: In if (tmymark$P1 == "AB" & tmymark$P2 == "AB") { :
  the condition has length > 1 and only the first element will be used
2: In if (tmymark$P1 == "AB" & tmymark$P2 == "BB") { :
  the condition has length > 1 and only the first element will be used

После того, как вектор loctype сформирован, я хочу перекодировать tmymark с информацией в этой переменной:

tmymark1 <- data.frame (loctype, tmymark)      
require(car) 
for(i in 2:length(tmymark)){

        if (loctype = "<hkxhk>") {
       tmymark[[i]] <- recode (x, "AB" = "hk", "BA" = "hk", "AA" = "hh", "BB" = "kk")
       } else {
       if (loctype = "<lmxll>") {
       tmymark[[i]] <- recode ((x, "AB" = "lm", "BA" = "lm", "AA" = "--", "BB" = "kk")
       } else {

        if (loctype = "<nnxnp>") {
       tmymark[[i]] <- recode ((x, "AB" = "np", "BA" = "np", "AA" = "nn", "BB" = "--")
             } else {
       if (loctype = "MN") {
        tmymark[[i]] <- "--"
       } esle {
      if (loctype = "NR") {
        tmymark[[i]] <- "NA"
       } else {
       cat ("error wrong input code")
      } } }}} 

Я на правильном пути?

Изменения: ожидаемый результат

      loctype  P1 P2 I1 I2 I3 I4 KL MN 
mark1  <lmxmm> lm mm lm mm mm lm -- mm 
mark2  <hkxhk> hk hk hh kk kk hh -- kk 
mark3 <nnxnp> nn np nn -- -- nn -- -- 
 and so on 

Ответы [ 2 ]

1 голос
/ 09 февраля 2012

match определенно путь. Я бы сделал два кадра данных в виде ключей, например:

key <- data.frame(
             P1=c("AB", "AB", "AA", "AA", "BB", "--", "AA"),
             P2=c("AB", "BB", "AB", "BB", "AA", "AA", "--"),
             loctype=c("<hkxhk>", "<lmxll>", "<nnxnp>", "MN", "MN", "NR", "NR"))

key2 <- cbind(
  `<hkxhk>` = c("hk","hk","hh","kk"),
  `<lmxll>` = c("lm", "lm", "--", "kk"),
  `<nnxnp>` = c("np", "np", "nn", "--"),
  MN = rep("--", 4),
  NR = rep("NA", 4) )
rownames(key2) = c("AB","BA", "AA", "BB")

, а затем используйте match на key1, чтобы получить loctype (как Джастин также рекомендует), а также на именах строк и столбцах key2, чтобы получить нужную замену, используя матричное индексирование для желаемое значение от ключа.

loctype <- key$loctype[match(with(tmymark, paste(P1, P2, sep="\b")), 
                             with(key, paste(P1, P2, sep="\b")))]
ii <- match(as.vector(as.matrix(tmymark)), rownames(key2))
jj <- rep(match(loctype, colnames(key2)), nrow(tmymark))
out <- as.data.frame(matrix(key2[cbind(ii,jj)], nrow=nrow(tmymark)))
colnames(out) <- colnames(tmymark)
rownames(out) <- rownames(tmymark)
out$loctype <- loctype

Результат выглядит следующим образом, где отсутствующие значения, потому что у меня нет значений для этих комбинаций в моих ключах.

> print(out, na="")
      P1 P2 I1 I2 I3 I4 KL MN loctype
mark1 lm kk lm kk kk lm    kk <lmxll>
mark2 hk hk hh kk kk hh    kk <hkxhk>
mark3                                
mark4 nn np nn -- -- nn    -- <nnxnp>
mark5 hk hk hh kk kk hh    kk <hkxhk>
mark6                                
mark7                                
mark8 -- -- -- -- -- --    --      MN
1 голос
/ 09 февраля 2012

Ваша первая ошибка возникает, если if хочет одно логическое значение (или выражение, которое оценивается как единое).Вместо этого вы можете использовать ifelse( ), который является «векторизованным» if

ifelse(tmymark$P1 == "AB" &  tmymark$P2 == "AB", loctype = "<hkxhk>", else clauses...)

Чтобы избежать длинной структуры if() else() (или как бы ifelse()), вы можете использовать match.создайте фрейм данных ваших ожидаемых комбинаций P1 и P2 и добавьте столбец для нужного типа:

matches <- data.frame(p1p2 = c('AB AB', 'AB BB', 'AA AB', 'AA BB', 'BB AA', '-- AA', 'AA --'),
                      loctype = c('<hkxhk>', '<lmxll>', '<nnxnp>', 'MN', 'MN', 'NR', 'NR'))
loctype <- matches$loctype[match(paste(tmymark$P1, tmymark$P2), matches$p1p2),]

Вторая часть может быть выполнена несколькими способами, но я рисую пробел на"аккуратный".

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...