如何使用purrr匹配查找表中的记录?

时间:2017-07-15 10:34:08

标签: r dplyr purrr

我有这个数据集

library(dplyr)
data_frame(Q1= c('AL', NA, 'TX', 'FL'), Q2=c('MN', 'CO', NA, NA), value=c(10,24,12,54)) 
# A tibble: 4 x 3
     Q1    Q2 value
  <chr> <chr> <dbl>
1    AL    MN    10
2  <NA>    CO    24
3    TX  <NA>    12
4    FL  <NA>    54

我正在尝试使用purrr使用查找表将Q1Q2中的值转换为完整的州名称

lktState <- data_frame(abb=state.abb, name=state.name)

到目前为止,我已尝试过此功能,但它无法正常工作

data_frame(Q1= c('AL', NA, 'TX', 'FL'), Q2=c('MN', 'CO', NA, NA), value=c(10,24,12,54)) %>% 
  mutate_at(vars('Q1','Q2'), purrr::map(.x = ., lktState$name[match(.x, lktState$abb)]))
  

匹配错误(.x,lktState $ abb):object&#39; .x&#39;找不到

2 个答案:

答案 0 :(得分:4)

基础R版本(可以进行矢量化,但这说明了这个概念):

xdf <- data.frame(
  Q1= c('AL', NA, 'TX', 'FL'),
  Q2 = c('MN', 'CO', NA, NA),
  value = c(10, 24, 12, 54),
  stringsAsFactors=FALSE
) -> xdf

xdf
##     Q1   Q2 value
## 1   AL   MN    10
## 2 <NA>   CO    24
## 3   TX <NA>    12
## 4   FL <NA>    54
lktState <- setNames(state.name, state.abb)

xdf$Q1 <- lktState[xdf$Q1]
xdf$Q2 <- lktState[xdf$Q2]

xdf
##        Q1        Q2 value
## 1 Alabama Minnesota    10
## 2    <NA>  Colorado    24
## 3   Texas      <NA>    12
## 4 Florida      <NA>    54

“tidyverse”

library(dplyr)

xdf <- data_frame(
  Q1= c('AL', NA, 'TX', 'FL'),
  Q2 = c('MN', 'CO', NA, NA),
  value = c(10, 24, 12, 54)
) -> xdf

xdf
## # A tibble: 4 x 3
##      Q1    Q2 value
##   <chr> <chr> <dbl>
## 1    AL    MN    10
## 2  <NA>    CO    24
## 3    TX  <NA>    12
## 4    FL  <NA>    54
lktState <- setNames(state.name, state.abb)

mutate_at(xdf, .vars=vars(-value), .funs=funs(lktState[.]))
## # A tibble: 4 x 3
##        Q1        Q2 value
##     <chr>     <chr> <dbl>
## 1 Alabama Minnesota    10
## 2    <NA>  Colorado    24
## 3   Texas      <NA>    12
## 4 Florida      <NA>    54

没有必要使用“apply”式惯用语来执行此基本查找表分配。

答案 1 :(得分:2)

我同意Sotos的观点,即加入是自然而然的方式。但是,您的purrr解决方案绝对可以修复。

你缺少三件事,

  1. 除了简单的单一功能之外,您需要在funs中使用mutate_at
  2. map函数对匿名函数使用~表示法。
  3. 您不想返回列表,而是返回字符向量,因此请使用_chr变体。
  4. mutate_at(df,
              vars('Q1', 'Q2'), 
              funs(purrr::map_chr(.x = ., ~lktState$name[match(.x, lktState$abb)])))
    

    给出:

    # A tibble: 4 x 3
           Q1        Q2 value
        <chr>     <chr> <dbl>
    1 Alabama Minnesota    10
    2    <NA>  Colorado    24
    3   Texas      <NA>    12
    4 Florida      <NA>    54
    

    数据

    df <- data_frame(Q1= c('AL', NA, 'TX', 'FL'), Q2=c('MN', 'CO', NA, NA), value=c(10,24,12,54))