如何管理不同维度数据框的列表（保留 \\Drop 列）？

Question

我有大量不同维度的数据框列表。 我的真实数据的维度太大，所以我立即创建了一个新列表（myList），如下所示：

myList <- list( data.frame(ID = c("T-02", "T-04","T-06"), 
                              Test = rnorm(3, mean=50, sd=10),Shape=c("C","S","r"), 
                              Time=rnorm(3, mean=90, sd=10),Event=c(1,0,1),
                              KPS=c(90,100,70),Sex=c("F","M","F"),Race=c("W","B","W")),
              
                data.frame(ID = c("T-02", "T-04","T-06"), Shape=c("C","S","r"),
                              Value = 1:3,Time=rnorm(3, mean=90, sd=10),
                              Event=c(1,0,1),
                              KPS=c(90,100,70),Sex=c("F","M","F"),Race=c("W","B","W")),
               data.frame(ID = c("T-02", "T-04","T-06"), 
                              Test = rnorm(3, mean=50, sd=10), 
                              Value = 1:3, Time=rnorm(3, mean=90, sd=10),Event=c(1,0,1),
                             KPS=c(90,100,70),Sex=c("F","M","F"),Race=c("W","B","W")),

                data.frame(ID = c("T-02", "T-04","T-06"), 
               Test = rnorm(3, mean=50, sd=10), 
               Value = 1:3,Time=rnorm(3, mean=90, sd=10),Event=c(1,0,1),
               KPS=c(90,100,70),Sex=c("F","M","F"),Race=c("W","B","W")))

我正在寻找一个函数，我可以在其中删除“事件”列之后的所有列，或者从第一列到“事件”列中选择列。 对于这么短的数据，我可以通过以下代码轻松完成：

new_drop<-lapply(myList, function(x){x[,!names(x)%in%c("KPS","Sex","Race")]})

但我想在我的数据中删除 20 多个这样的列。 我想知道是否有更简单的方法。 我也试过这段代码，但它不能正常工作！

new_drop1<-lapply(myList, function(x){x[,endsWith(colnames(x),"Event")]})

我很感激任何帮助。

Answer 1

下面的代码允许您仅将前面的列保留为“事件”，使用which 。 which是一个base function ，它返回满足一组条件的索引。 这里唯一的条件是colnames(x)=="Event" 。

new_drop<-lapply(myList, function(x){
  res <- NULL
  if(is.data.frame(x)) {
    eventcol <- which(colnames(x)=="Event")
    res <- x[,1:eventcol]
  } else {
    res <- x
  }
  return(res)
         })

即使不是所有的list元素都属于data.frame class这也会起作用

> lapply(new_drop, head)
[[1]]
    ID     Test Shape     Time Event
1 T-02 57.38475     C 76.05545     1
2 T-04 40.84934     S 85.98049     0
3 T-06 45.44281     r 85.18336     1

[[2]]
    ID Shape Value      Time Event
1 T-02     C     1 101.68492     1
2 T-04     S     2 100.13524     0
3 T-06     r     3  89.14877     1

[[3]]
    ID     Test Value     Time Event
1 T-02 42.92581     1 82.37073     1
2 T-04 42.10800     2 90.51706     0
3 T-06 50.51329     3 96.52649     1

[[4]]
    ID     Test Value     Time Event
1 T-02 49.13385     1 85.91036     1
2 T-04 52.72536     2 98.83747     0
3 T-06 68.96858     3 96.51575     1

Answer 2

你可以使用grep ：

lapply(myList, function(x) x[seq(grep("Event", names(x)))])


[[1]]
    ID     Test Shape     Time Event
1 T-02 65.11001     C 94.53361     1
2 T-04 70.25636     S 84.86061     0
3 T-06 44.56480     r 85.30492     1

[[2]]
    ID Shape Value     Time Event
1 T-02     C     1 93.40279     1
2 T-04     S     2 85.78726     0
3 T-06     r     3 97.02140     1

[[3]]
    ID     Test Value     Time Event
1 T-02 39.89387     1 94.80438     1
2 T-04 48.28122     2 85.62445     0
3 T-06 49.47685     3 90.10609     1

[[4]]
    ID     Test Value     Time Event
1 T-02 38.55385     1 78.33900     1
2 T-04 47.60908     2 77.63453     0
3 T-06 43.59754     3 92.25645     1

Answer 3

对于每个数据框，您可以使用ncol查找列数，然后使用which函数查找事件列的位置。 例如，对于 mtcars，假设 wt 列是“事件”列。

cars <- mtcars
ncols <- ncol(cars)
11
wt <- which(colnames(cars) == "wt")
6

# To remove columns 1 to to wt-1 and wt+1 to end

before <- wt - 1
cars1 <- cars[ , -(1:before)]
after <- wt + 1
cars2 <- cars[ , -(after:ncols)]

您可以使用sapply函数来查找所有数据框的维度和事件列的位置，然后使用lapply来处理列删除。