which函数的用法

求教高手

有两个数据框，

其中A格式为

中国、美国                1
德国                          6
日本、朝鲜                10

B数据格式为

中国                  
德国、法国      
泰国
美国

想要在B框中，加上一列，值为A框中的数字，条件是：A中的国家名包含B中该行的国家名，或B中该行的国家名包含A中的国家名

得到结果为


中国             1
德国、法国   6
泰国            NA
美国             1

这个东西用which比较难写， 而且直接在这两个data frame间比较，逻辑很烦。我把第一个data frame重新调整了一下来做，逻辑上好写。 dplyr的pipeline一直会出错，所以用了很多的for循环，但应该可以得到你要的结果

1. library(dplyr)
   
2. 
   
3. a=data.frame(Cntry=c('中国、美国','德国','日本、朝鲜'),
   
4.              Val=c(1,6,10),
   
5.              stringsAsFactors=F)
   
6. 
   
7. b=data.frame(Cntry=c('中国','德国、法国','泰国','美国'),
   
8.              Val=NA,
   
9.              stringsAsFactors=F)
   
10. 
    
11. # Function to get the value in a by country name--------------
    
12. GetVal=function(Cntry)
    
13. {
    
14.   Val=NA
    
15.   for (i in 1:nrow(a))
    
16.   {
    
17.     x=a$Cntry[i] %>%
    
18.       strsplit(.,split='、') %>%
    
19.       unlist
    
20.     if(Cntry %in% x) Val=a$Val[i]
    
21.   }
    
22.   return(Val)
    
23. }
    
24. 
    
25. # Get an organized data frame -------
    
26. a$Cntry %>%
    
27.   strsplit(,split='、') %>%
    
28.   unlist %>%
    
29.   data.frame(Cntry_NM=.,Val=NA,
    
30.              stringsAsFactors=F) -> base
    
31. 
    
32. 
    
33. for(i in 1:nrow(base))
    
34. {
    
35.   base$Val[i]=GetVal(base$Cntry_NM[i])
    
36. }
    
37. 
    
38. for(i in 1:nrow(b))
    
39. {
    
40.   Ind=base$Cntry_NM %in% unlist(strsplit(b$Cntry,split='、')[i])
    
41.   if(sum(Ind)>0) b$Val[i]=base$Val[Ind]
    
42. }

复制代码