【问题标题】:How to apply a function that works with data.frame cells to data.frame columns如何将适用于 data.frame 单元格的函数应用于 data.frame 列
【发布时间】:2015-01-18 20:51:53
【问题描述】:

这个问题是对我之前的一个问题的改编,我觉得我问的方式不清楚。我正在检查列 V1 和 V2 是否按行有共同代码。代码由正斜杠“/”分隔。下面的函数应该在同一行从 V1 中获取一个单元格,从 V2 中获取一个单元格,并将它们转换为向量。向量的每个元素都是一个代码。然后该函数应检查获得的两个向量是否具有共同的元素。这些元素最初是 4 位代码。如果两个向量之间有任何 4 位代码匹配,则该函数应返回 4。如果没有共同的元素,该函数应减少每个代码的位数,然后再次检查。每次该函数减少位数时,它也会减少最后返回的分数。我希望将函数返回的值写入我选择的列中。

这是我的起始条件

structure(list(ID = c(2630611040, 2696102020, 2696526020), V1 = c("7371/3728", 
"2834/2833/2836/5122/8731", "3533/3541/3545/5084"), V2 = c("7379", 
"3841", "3533/3532/3531/1389/8711")), .Names = c("ID", "V1", 
"V2"), class = "data.frame", row.names = c(NA, 3L))

         ID                       V1                       V2
1 2630611040                7371/3728                     7379
2 2696102020 2834/2833/2836/5122/8731                     3841
3 2696526020      3533/3541/3545/5084 3533/3532/3531/1389/8711

我想得到这个

          ID                       V1                       V2   V3
1 2630611040                7371/3728                     7379   3
2 2696102020 2834/2833/2836/5122/8731                     3841   0
3 2696526020      3533/3541/3545/5084 3533/3532/3531/1389/8711   4

我的功能是这个

coderelat<-function(a, b){

a<-unique(as.integer(unlist(str_split(a, "/")))) #Transforming cells into vectors of codes
b<-unique(as.integer(unlist(str_split(b, "/"))))

a<-a[!is.na(a)]
b<-b[!is.na(b)]

if (length(a)==0 | length(b)==0) { # Check that both cells are not empty

  ir=NA     
  return(ir)

  } else {


for (i in 3:1){

    diff<-intersect(a, b) # See how many products the shops have in common

            if (length(diff)!=0) { #As you find a commonality, give ir the corresponding scoring

              ir=i+1
              break

            } else if (i==1 & length(diff)==0) { #If in the last cycle, there is still no commonality put ir=0

              ir=0
              break

            } else { # If there is no commonality and you are not in the last cycle, reduce the nr. of digits and re-check commonality again

              a<- unique(as.integer(substr(as.character(a), 1, i)))
              b<- unique(as.integer(substr(as.character(b), 1, i)))

        }

    }     
  }
return(ir)
}

当我手动提供单个单元格时,该功能有效。但是当我写这样的东西时它不起作用:

df$V4<-coderelat(df$V1, df$V2)

我非常感谢任何帮助,因为我不再知道如何进行这项工作。

提前非常感谢。 里卡多

【问题讨论】:

  • 使用 dput(...) 提供您的数据非常有帮助 (+1)。

标签: r function dataframe


【解决方案1】:

这是一个使用 data.tables 的解决方案。

get.match <-function(a,b) {
  A <- unique(strsplit(a,"/",fixed=TRUE)[[1]])
  B <- unique(strsplit(b,"/",fixed=TRUE)[[1]])
  for (i in 4:1) if(length(intersect(substr(A,1,i),substr(B,1,i)))>0) return(i)
  return(0L)
}
library(data.table)
setDT(df)[,V3:=get.match(V1,V2),by=ID]
df
#            ID                       V1                       V2 V3
# 1: 2630611040                7371/3728                     7379  3
# 2: 2696102020 2834/2833/2836/5122/8731                     3841  0
# 3: 2696526020      3533/3541/3545/5084 3533/3532/3531/1389/8711  4

【讨论】:

  • 这是一个很好的答案!非常感谢,真的。我有两个问题要澄清你的解决方案。首先,您能否解释一下 get.match 的第 2 行和第 3 行末尾的 [[1]] 吗?其次,如果A = NA,该函数会做什么?再次感谢!
  • strsplit(...) 创建一个字符向量列表。在您的情况下,该列表只有一个元素,因此我们提取的是使用 [[1]]
猜你喜欢
  • 1970-01-01
  • 2018-07-17
  • 2015-02-23
  • 2013-01-27
  • 2016-03-20
  • 2017-01-16
  • 1970-01-01
  • 1970-01-01
  • 1970-01-01
相关资源
最近更新 更多