Chọn nhiều cột trong data.table bằng các chỉ số số của chúng

Làm thế nào chúng ta có thể chọn nhiều cột bằng cách sử dụng một vectơ chỉ số (vị trí) của chúng trong data.table?

Đây là cách chúng tôi sẽ làm với data.frame:

df <- data.frame(a = 1, b = 2, c = 3)
df[ , 2:3]
#   b c
# 1 2 3

Đối với các phiên bản của data.table >= 1.9.8 , tất cả những thứ sau đây chỉ hoạt động:

library(data.table)
dt <- data.table(a = 1, b = 2, c = 3)

# select single column by index
dt[, 2]
#    b
# 1: 2

# select multiple columns by index
dt[, 2:3]
#    b c
# 1: 2 3

# select single column by name
dt[, "a"]
#    a
# 1: 1

# select multiple columns by name
dt[, c("a", "b")]
#    a b
# 1: 1 2

Đối với các phiên bản data.table < 1.9.8 (trong đó lựa chọn cột số yêu cầu sử dụng with = FALSE), hãy xem phiên bản trước của câu trả lời này. Xem thêm TIN TỨC trên v1.9.8, TIỀM NĂNG THAY ĐỔI, điểm 3.

Hơi dài dòng một chút, nhưng tôi đã quen với việc sử dụng .SDbiến ẩn .

b<-data.table(a=1,b=2,c=3,d=4)
b[,.SD,.SDcols=c(1:2)]

Hơi rắc rối một chút, nhưng bạn không mất đi các tính năng dữ liệu khác (tôi không nghĩ vậy), vì vậy bạn vẫn có thể sử dụng các chức năng quan trọng khác như bảng tham gia, v.v.

Nếu bạn muốn sử dụng tên cột để chọn các cột, chỉ cần sử dụng .(), đó là bí danh cho list():

library(data.table)
dt <- data.table(a = 1:2, b = 2:3, c = 3:4)
dt[ , .(b, c)] # select the columns b and c
# Result:
#    b c
# 1: 2 3
# 2: 3 4

Từ v1.10.2 trở đi, bạn cũng có thể sử dụng ..

dt <- data.table(a=1:2, b=2:3, c=3:4)

keep_cols = c("a", "c")

dt[, ..keep_cols]

@Tom, cảm ơn bạn rất nhiều vì đã chỉ ra giải pháp này. Đó là công việc tốt cho tôi.

Tôi đang tìm cách để loại trừ một cột khỏi in và từ ví dụ trên. Để loại trừ cột thứ hai, bạn có thể làm một cái gì đó như thế này

library(data.table)
dt <- data.table(a=1:2, b=2:3, c=3:4)
dt[,.SD,.SDcols=-2]
dt[,.SD,.SDcols=c(1,3)]