Как удалить столбец по имени в data.table?

Чтобы избавиться от столбца с именем "foo" в a data.frame, я могу сделать:

df <- df[-grep('foo', colnames(df))]

Однако после dfпреобразования в data.tableобъект просто невозможно удалить столбец.

Пример:

df <- data.frame(id = 1:100, foo = rnorm(100))
df2 <- df[-grep('foo', colnames(df))] # works
df3 <- data.table(df)
df3[-grep('foo', colnames(df3))] 

Но как только он превращается в data.tableобъект, он больше не работает.

Любое из следующего удалит столбец fooиз data.table df3:

# Method 1 (and preferred as it takes 0.00s even on a 20GB data.table)
df3[,foo:=NULL]

df3[, c("foo","bar"):=NULL]  # remove two columns

myVar = "foo"
df3[, (myVar):=NULL]   # lookup myVar contents

# Method 2a -- A safe idiom for excluding (possibly multiple)
# columns matching a regex
df3[, grep("^foo$", colnames(df3)):=NULL]

# Method 2b -- An alternative to 2a, also "safe" in the sense described below
df3[, which(grepl("^foo$", colnames(df3))):=NULL]

data.table также поддерживает следующий синтаксис:

## Method 3 (could then assign to df3, 
df3[, !"foo"]  

хотя, если вы действительно хотите удалить столбец "foo"из df3(а не просто печатать представление df3минус столбца "foo"), вы бы действительно хотели использовать метод 1.

(Обратите внимание, что если вы используете метод, основанный на grep()или grepl(), вам нужно установить pattern="^foo$"вместо "foo", если вы не хотите, чтобы столбцы с такими именами, как "fool"и "buffoon"(то есть, содержащие fooв качестве подстроки), также сопоставлялись и удалялись.)

Менее безопасные варианты, отлично подходит для интерактивного использования:

Следующие две идиомы также будут работать - если df3содержит сопоставление столбцов"foo" - но потерпят неудачу, вероятно, неожиданным образом, если это не так. Если, например, вы используете какой-либо из них для поиска несуществующего столбца "bar", вы получите нулевую строку data.table.

Как следствие, они действительно лучше всего подходят для интерактивного использования, когда можно, например, захотеть отобразить таблицу данных за вычетом любых столбцов с именами, содержащими подстроку "foo". Для целей программирования (или если вы действительно хотите удалить столбец (столбцы), df3а не из его копии), методы 1, 2a и 2b - действительно лучшие варианты.

# Method 4:
df3[, .SD, .SDcols = !patterns("^foo$")]

Наконец, есть подходы, использующие подходы with=FALSE, хотя они data.tableпостепенно отходят от использования этого аргумента, поэтому сейчас не рекомендуется, где вы можете его избежать; показывая здесь, чтобы вы знали, что опция существует на тот случай, если она вам действительно нужна:

# Method 5a (like Method 3)
df3[, !"foo", with=FALSE] 
# Method 5b (like Method 4)
df3[, !grep("^foo$", names(df3)), with=FALSE]
# Method 5b (another like Method 4)
df3[, !grepl("^foo$", names(df3)), with=FALSE]

Вы также можете использовать setэто, чтобы избежать накладных расходов [.data.tableв циклах:

dt <- data.table( a=letters, b=LETTERS, c=seq(26), d=letters, e=letters )
set( dt, j=c(1L,3L,5L), value=NULL )
> dt[1:5]
   b d
1: A a
2: B b
3: C c
4: D d
5: E e

Если вы хотите сделать это по имени столбца, which(colnames(dt) %in% c("a","c","e"))должно работать j.

Я просто делаю это во фрейме данных таким образом:

DT$col = NULL

Работает быстро и, насколько я вижу, проблем не вызывает.

ОБНОВЛЕНИЕ: не лучший метод, если ваш DT очень большой, так как использование $<-оператора приведет к копированию объекта. Так что лучше используйте:

DT[, col:=NULL]

Очень простой вариант, если у вас есть много отдельных столбцов для удаления в таблице данных, и вы хотите избежать ввода всех имен столбцов #careadviced

dt <- dt[, -c(1,4,6,17,83,104)]

Вместо этого будут удалены столбцы на основе номера столбца.

Очевидно, что он не так эффективен, потому что он обходит преимущества data.table, но если вы работаете с менее чем 500 000 строк, он работает нормально

Предположим , что ваш дт имеет столбцы col1, col2, col3, col4, col5, coln.

Чтобы удалить их подмножество:

vx <- as.character(bquote(c(col1, col2, col3, coln)))[-1]
DT[, paste0(vx):=NULL]

Вот способ, когда вы хотите установить количество столбцов в NULL, если их имена столбцов являются функцией для вашего использования :)

deleteColsFromDataTable <- function (train, toDeleteColNames) {

       for (myNm in toDeleteColNames)

       train <- train [,(myNm):=NULL]

       return (train)
}

DT[,c:=NULL] # remove column c

Для data.table назначение столбца в NULL удаляет его:

DT[,c("col1", "col1", "col2", "col2")] <- NULL
^
|---- Notice the extra comma if DT is a data.table

... что эквивалентно:

DT$col1 <- NULL
DT$col2 <- NULL
DT$col3 <- NULL
DT$col4 <- NULL

Эквивалент для data.frame:

DF[c("col1", "col1", "col2", "col2")] <- NULL
      ^
      |---- Notice the missing comma if DF is a data.frame

В. Почему в версии для data.table есть запятая, а в версии data.frame нет запятой?

О. Поскольку data.frames хранятся в виде списка столбцов, вы можете пропустить запятую. Вы также можете добавить его, тем не менее , то вам нужно будет назначить их в список NULLс, DF[, c("col1", "col2", "col3")] <- list(NULL).