通过名称或索引引用列的功能

时间:2019-07-18 19:13:52

标签: r string function vector numeric

我希望能够在一个向量中通过名称和索引来引用列。例如,我仅指定:

编辑:由于我希望顺序无关紧要,所以我更改了原始矢量的顺序。

columns <- c(1:7, "j", 8, "i")

然后,我想检索索引1到9的名称,并将它们添加到向量中(在正确的位置)。我有一个大致的想法,但是在编码方面我走得并不远:

library(data.table)
df <- fread(
"a b c d e f g h i j
1 2 3 4 5 6 7 8 9 10",
  header = TRUE
)

function(data, columns){
nums <- as.numeric(columns)
named_columns <- ?
nums <- nums[!is.na(nums)]
name_nums <- colnames(df)[nums]

all_columns <- setdiff(named_colums, name_nums)
# Order of the original vector?
column_names <- result
}

,然后将其返回到向量,以使结果为:

column_names <- c("a", ..., "j", "h", "i")

有人可以帮助我进一步发展吗?

1 个答案:

答案 0 :(得分:2)

如果我理解您的问题,那么您想传递包含列名和索引的向量,并且只想返回包含列索引的向量。然后以下内容会有所帮助;

df <- data.table::fread("a b c d e f g h i j
                         1 2 3 4 5 6 7 8 9 10",
                                               header = TRUE)
columns <- c(1:8, "i", 9, "j")


col2num <- function(df, columns){
              nums <- as.numeric(columns)
              nums[is.na(nums)] <- which(names(df)==columns[is.na(nums)])
              return(nums)
            }

col2num(df, columns)
#> Warning in col2num(df, columns): NAs introduced by coercion
#>  [1]  1  2  3  4  5  6  7  8  9  9 10