r - 如何以编程方式在数据框中分配一堆变量

Question

给定一个字符向量 vars 和一个数据帧列表 d，我想确保 d 中的每个数据框都包含 vars 中命名的所有列。假设一个数据框中缺少一些列，然后我创建数据框中的那些列并用 NA 填充它们。

但是，当我使用 assign 执行此操作时，我得到了一些奇怪的结果:

> vars <- c('y','z')
> b <- data.frame(a=1:3, b=3:1)
> b
  a b
1 1 3
2 2 2
3 3 1
> within(b, for (v in vars) assign(v, NA))
  a b  z  y v
1 1 3 NA NA z
2 2 2 NA NA z
3 3 1 NA NA z

您可以看到我使用此方法成功创建了列 z 和 y，但是还有一个额外的卷 v 我不知道它是从哪里来的。

Answer 1

这是一种符合原始代码精神的简单方法。

for(v in vars) { b[[v]] <- NA }

在您的版本中获得额外 v 的原因是在调用 within 时创建的任何变量都会添加到该数据框中，并且for 循环创建该变量。如果你在最后删除它，它就会消失。但是，请注意您的 vars 变量包含 v 的情况。

within(b, {for (v in vars) assign(v, NA); rm(v) })

您也可以让 vars 包含所有变量，然后使用 setdiff 获取您想要保留的变量。

vars <- c('a','b','y','z')
b <- data.frame(a=1:3, b=3:1)
for(v in setdiff(vars, names(b))) { b[[v]] <- NA }