使用 R 选择指定列之后的所有列答案

【问题标题】：select all columns after a designated column using R使用 R 选择指定列之后的所有列
【发布时间】：2021-09-24 12:34:14
【问题描述】：

如何使用 R 选择指定列之后的所有列（理想情况下仅限 dplyr，但欢迎非 dplyr 解决方案）。例如，假设在数据框mtcars 中，我想抓取vs 之后的所有列，即am gear carb。但我想要一个函数来完成此任务，其中我只需提供选择的列，返回的数据框仅包括指定列之后的那些列。我想不出一种方法来完成这项工作。

                     mpg cyl  disp  hp drat    wt  qsec vs am gear carb
Mazda RX4           21.0   6 160.0 110 3.90 2.620 16.46  0  1    4    4
Mazda RX4 Wag       21.0   6 160.0 110 3.90 2.875 17.02  0  1    4    4
Datsun 710          22.8   4 108.0  93 3.85 2.320 18.61  1  1    4    1
Hornet 4 Drive      21.4   6 258.0 110 3.08 3.215 19.44  1  0    3    1
Hornet Sportabout   18.7   8 360.0 175 3.15 3.440 17.02  0  0    3    2
Valiant             18.1   6 225.0 105 2.76 3.460 20.22  1  0    3    1
Duster 360          14.3   8 360.0 245 3.21 3.570 15.84  0  0    3    4

我希望函数的结果如下（但我对结果的兴趣远低于逻辑，因为结果可以通过多种方式获得，但逻辑可推广到复杂情况（大量列，更改参数等）。

                    am gear carb
Mazda RX4            1    4    4
Mazda RX4 Wag        1    4    4
Datsun 710           1    4    1
Hornet 4 Drive       0    3    1
Hornet Sportabout    0    3    2
Valiant              0    3    1
Duster 360           0    3    4

【问题讨论】：

对不起，我应该更清楚一点，不一定是用户定义的函数，而是 dplyr 的函数/动词库中的东西）。

标签： r select dplyr sparklyr

【解决方案1】：

一个选项可能是：

mtcars %>%
 select(vs:last_col(), -vs)

                    am gear carb
Mazda RX4            1    4    4
Mazda RX4 Wag        1    4    4
Datsun 710           1    4    1
Hornet 4 Drive       0    3    1
Hornet Sportabout    0    3    2
Valiant              0    3    1
Duster 360           0    3    4
Merc 240D            0    4    2
Merc 230             0    4    2

或者：

mtcars %>%
 select((which(names(.) == "vs") + 1):last_col())

【讨论】：

谢谢，last_col() 正是我想要的。

【解决方案2】：

在基础 R 中你可以这样做：

my_select <- function(data, col_after){
   data[-seq_len(grep(col_after, names(data)))]
}

my_select(head(mtcars), "vs")
                  am gear carb
Mazda RX4          1    4    4
Mazda RX4 Wag      1    4    4
Datsun 710         1    4    1
Hornet 4 Drive     0    3    1
Hornet Sportabout  0    3    2
Valiant            0    3    1

在 dplyr 中你会这样做：

my_select1 <- function(data, col_after){
  data %>%
    select(!1:{{col_after}})
}

my_select1(head(mtcars), vs)
                 am gear carb
Mazda RX4          1    4    4
Mazda RX4 Wag      1    4    4
Datsun 710         1    4    1
Hornet 4 Drive     0    3    1
Hornet Sportabout  0    3    2
Valiant            0    3    1

【讨论】：

【解决方案3】：

你可以先找到vs的索引，然后像这样使用last_col：

library(tidyverse)
vs_index = which(colnames(mtcars)=="vs")
mtcars %>% 
        select((vs_index + 1):last_col())

【讨论】：

【解决方案4】：

您可以为此使用匹配：

mtcars %>%
  select((match("vs", names(mtcars))+1):ncol(mtcars))

【讨论】：