繁体   English   中英

子集数据帧并使用loop或lapply存储到R中的不同变量

[英]Subset a data frame and store to different variables in R using loop or lapply

我有一个数据框架,我想对其进行几次子集并将其存储在不同的变量名称中。 假设我的数据框看起来像这样:

set.seed(123)
x <- rnorm(5)
y <- rnorm(5)
z <- rnorm(5)

f1 <- gl(2,1, labels = c("good", "bad"), length =5)
f2 <- gl(3,1, labels = c("red", "green", "yellow"), length = 5)
f3 <- gl(5,1, labels = c("foo", "bar", "foobar", "foofoo", "barbar"))

df <- data.frame(x,y,z,f1,f2,f3)    
> df

            x          y          z   f1     f2     f3
1 -0.56047565  1.7150650  1.2240818 good    red    foo
2 -0.23017749  0.4609162  0.3598138  bad  green    bar
3  1.55870831 -1.2650612  0.4007715 good yellow foobar
4  0.07050839 -0.6868529  0.1106827  bad    red foofoo
5  0.12928774 -0.4456620 -0.5558411 good  green barbar

我想做的是通过设置df来创建三个新的数据帧,并将它们存储到不同的变量名中。 我知道如何单独执行此操作:

df_f1 <- df[,c(-5,-6)]

> df_f1
            x          y          z   f1
1 -0.56047565  1.7150650  1.2240818 good
2 -0.23017749  0.4609162  0.3598138  bad
3  1.55870831 -1.2650612  0.4007715 good
4  0.07050839 -0.6868529  0.1106827  bad
5  0.12928774 -0.4456620 -0.5558411 good

df_f2 <- df[,c(-4,-6)]

> df_f2
            x          y          z     f2
1 -0.56047565  1.7150650  1.2240818    red
2 -0.23017749  0.4609162  0.3598138  green
3  1.55870831 -1.2650612  0.4007715 yellow
4  0.07050839 -0.6868529  0.1106827    red
5  0.12928774 -0.4456620 -0.5558411  green

df_f3 <- df[,c(-4,-5)]
> df_f3
            x          y          z     f3
1 -0.56047565  1.7150650  1.2240818    foo
2 -0.23017749  0.4609162  0.3598138    bar
3  1.55870831 -1.2650612  0.4007715 foobar
4  0.07050839 -0.6868529  0.1106827 foofoo
5  0.12928774 -0.4456620 -0.5558411 barbar

但是,有没有办法以编程方式进行此操作? 也许使用for循环或套用? 我的问题是我不知道如何自动将我需要的数据帧分配给不同的变量名称,例如df_f1,df_f2和df_f3,而无需手动手动键入它们。 我的意思是,有没有一种方法可以自动生成变量名,以便可以使用循环或lapply在它们上存储数据帧?

我将这个概念应用于更大的数据集,并且手动键入每个变量名称非常繁琐。

谢谢,祝大家有美好的一天!

list2env(setNames(lapply(df[-(1:3)],cbind,df[1:3]),paste("df",1:3,sep="_f")),.GlobalEnv)

分解:

首先创建一个需要的列表,其中包含所有数据框。

  A=lapply(df[-(1:3)],cbind,df[1:3])

这将所有其他列从1:3开始,然后将每个列与df[1:3]绑定。 这给了我一个列表A,其中包含我需要的所有数据帧。 现在给列表中的每个数据框一个名称:

  B=setNames(A,paste("df",1:3,sep="_f"))

您可以玩paste ,看看它如何将两件事结合在一起。 之后。 我们将列出列表中的每个元素,从技术上讲,这是我们全球环境的数据框架。

 list2env(B,.GlobalEnv)

这似乎有效,使用lapply

keep<-3
split_id<-(keep+1):length(df)
df_list<- lapply(split_id, function(x){
  df[,c(1:3,x)]
})

df_list
[[1]]
            x          y          z   f1
1 -0.56047565  1.7150650  1.2240818 good
2 -0.23017749  0.4609162  0.3598138  bad
3  1.55870831 -1.2650612  0.4007715 good
4  0.07050839 -0.6868529  0.1106827  bad
5  0.12928774 -0.4456620 -0.5558411 good

[[2]]
            x          y          z     f2
1 -0.56047565  1.7150650  1.2240818    red
2 -0.23017749  0.4609162  0.3598138  green
3  1.55870831 -1.2650612  0.4007715 yellow
4  0.07050839 -0.6868529  0.1106827    red
5  0.12928774 -0.4456620 -0.5558411  green

[[3]]
            x          y          z     f3
1 -0.56047565  1.7150650  1.2240818    foo
2 -0.23017749  0.4609162  0.3598138    bar
3  1.55870831 -1.2650612  0.4007715 foobar
4  0.07050839 -0.6868529  0.1106827 foofoo
5  0.12928774 -0.4456620 -0.5558411 barbar

你的意思是这样吗?

dependent_col = c("f1", "f2", "f3")
df_l <- lapply(dependent_col, function(x) df[!(colnames(df) %in% dependent_col) | colnames(df) == x])
names(df_l) <- paste("df", dependent_col, sep="_")
df_l

您可以使用df_l$df_f1df_l$df_f2等访问单个数据帧...

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM