访问lapply列名

Question

如果我在做

lapply(dataframe, function(x) {
    column.name <- #insert code here
})

如何访问lapply函数当前正在处理的列的名称？如代码所示，我希望将列名分配给变量column.name。为了澄清，是的，column.name将随着lapply的每一次迭代而改变。

Answer 1

其实是有办法的。

df <- data.frame(a = 1:2, b = 3:4, c = 5:6)
lapply(df, function(x) names(df)[substitute(x)[[3]]])
$a
[1] "a"

$b
[1] "b"

$c
[1] "c"

但这应该作为最后的手段。相反，使用类似的东西(注释中给出了另一个选项)

lapply(seq_along(df), function(x) names(df[x]))
[[1]]
[1] "a"

[[2]]
[1] "b"

[[3]]
[1] "c"

Answer 2

您可以迭代索引，但这不是一个类似R的代码。一种更直接的方法是使用Map，这是lapply的多元版本，它在传递给它的任何参数中并行地迭代一个适当的函数：

Map(function(value, name){paste(name, sum(value), sep = ": ")}, 
    Formaldehyde, 
    names(Formaldehyde))
#> $carb
#> [1] "carb: 3.1"
#> 
#> $optden
#> [1] "optden: 2.747"

如果使用tidyverse，purrr::imap是类似的方便版本的purrr::map2，它自动使用第一个参数的名称作为第二个参数：

purrr::imap(Formaldehyde, ~paste(.y, sum(.x), sep = ": "))
#> $carb
#> [1] "carb: 3.1"
#> 
#> $optden
#> [1] "optden: 2.747"

每个简化版本都是可用的:对于Map，mapply，一个多变量sapply (从技术上讲，Map只是一个带有SIMPLIFY = FALSE的包装器)；对于imap，有一个要简化的类型下标的版本，例如imap_chr。

Answer 3

How to pass a variable into the function while using lapply

一个带有两个变量的lapply，所以我不必一直重写每个状态的函数。

library(tidycensus)    
get_Census <- function(x,y) {
      get_decennial(geography = "block group",
                    variables = "P001001",
                    sumfile = "sf1",
                    key = mykey,
                    state = x, county = y,year = "2000",
                    geometry = FALSE)
    }
    CO<-c("067","073","113")
    lapply(CO,get_Census,x="06")