选择列,如果列不存在则跳过

Question

我想通过选择列来制作我的数据的子集,如下所示:

select(df, col1, col2, col3, col4) 

但有时我的数据集略有不同,只有col1,col2和col4.

如何使用select(),如果列不存在,它会继续而不会出错？

因此它会给出一个带col1,col2和col4的数据集(并跳过col3).如果我只运行上面的select()行,我会收到此错误:

Error in overscope_eval_next(overscope, expr) : object 'col3' not found

Answer 1

df[, names(df) %in% c('col1', 'col2', 'col3', 'col4')]

Answer 2

您可以使用dplyr 中的one_of() 选择助手并将列名作为字符串传递。它只会对不存在的列发出警告。

library(dplyr)

select(mtcars, one_of(c("mpg", "disp", "foo")))

#> Warning: Unknown variables: `foo`

#>                      mpg  disp
#> Mazda RX4           21.0 160.0
#> Mazda RX4 Wag       21.0 160.0
#> Datsun 710          22.8 108.0
#> Hornet 4 Drive      21.4 258.0
#> Hornet Sportabout   18.7 360.0