Выбор столбцов из data.table

Работа с данными с помощью data.table в R

Matt Dowle, Arun Srinivasan

Instructors, DataCamp

Общий синтаксис data.table (повторение)

Второй аргумент j используется для выбора столбцов (и вычислений над ними)

# General form of data.table syntax
DT[i, j, by]
   |  |  |
   |  |  --> grouped by what?
   |  -----> what to do?
   --------> on which rows?

Выбор столбцов по имени

Аргумент j принимает символьный вектор с именами столбцов

ans <- batrips[, c("trip_id", "duration")]
head(ans, 2)

trip_id duration
139545      435
139546      432

Выбор столбцов по имени

batrips_df <- as.data.frame(batrips)
ans <- batrips_df[, "trip_id"]
head(ans, 2)

# The result is a vector, not a data.frame
139545, 139546

ans <- batrips[, "trip_id"]
# Still a data.table, not a vector
head(ans, 2)

trip_id
139545
139546

Вместо имён можно использовать номера столбцов

ans <- batrips[, c(2, 4)]
head(ans, 2)

duration  start_station
435       San Francisco City Hall
432       San Francisco City Hall

Однако это считается плохой практикой

# If the order of columns changes, the result is wrong
batrips[, c(2, 4)]

# The result is always correct, no matter the order
batrips[, c("duration", "start_station")]

Исключение столбцов с помощью символьных векторов

-c("col1", "col2", ...) исключает указанные столбцы
Это удобная функция, доступная только в data.table
Вместо - можно использовать ! — результат тот же

# Select all cols *except* those shown below
ans <- batrips[, -c("start_date", "end_date", "end_station")]
head(ans, 1)

trip_id  duration  start_station             start_terminal  bike_id  end_terminal
139545   435       San Francisco City Hall   58              65       473 

subscription_type  zip_code
Subscriber         94612

Выбор столбцов в стиле data.table

Помните, как столбцы использовались как переменные в аргументе i в прошлой главе?

# Recap the "i" argument
# All trips more than an hour
batrips[duration > 3600]

Аналогично можно использовать список переменных (имён столбцов) для выбора столбцов

ans <- batrips[, list(trip_id, dur = duration)]
head(ans, 2)

trip_id     dur
139545      435
139546      432

Если выбирается один столбец без обёртки list(), результатом будет vector

# Select a single column and return a data.table
ans <- batrips[, list(trip_id)]
head(ans ,2)

trip_id
139545
139546

# Select a single column and return a vector
ans <- batrips[, trip_id]
head(ans, 2)

139545 139546

Выбор столбцов в стиле data.table

.() — это псевдоним list(), удобный для краткости

# .() is the same as list()
ans <- batrips[, .(trip_id, duration)]
head(ans, 2)

trip_id duration
139545      435
139546      432

Давайте потренируемся!

Работа с данными с помощью data.table в R