Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Create vector of data frame names from a list of data frames

Tags:

list

dataframe

r

I have a list of data frames, and I want to create a character vector that contains the data frame names that meet a certain criteria. In this case, I want the names of the data frames that contain at least 1 observation. I need these names in a vector for use in another function later on.

The list of data frames is created automatically by the split function. For reproducibility, here's a list of data frames:

df1 <- data.frame(replicate(2,sample(0:1,5,rep=TRUE)))
df2 <- data.frame(replicate(2,sample(0:1,5,rep=TRUE)))
df3 <- data.frame(replicate(0,sample(0:1,5,rep=TRUE))) #empty

mylist <- list(df1, df2,df3) #create list of data frames

mylist2 <- mylist[sapply(mylist, function(x) dim(x)[1]) > 0] #subset list so that df's with no observations are dropped

Now, what I need is a character vector with the names of the data frames in mylist2. (In this example, "df1" and "df2".)

Thanks!

like image 702
user3780463 Avatar asked Jul 06 '17 15:07

user3780463


2 Answers

If you want this to be somewhat automatized, for instance you have a large number of data.frames in your environment, you might want to select them with a pattern matching in one line:

mylist = do.call("list", mget(grep("df", ls(), value=T)))
mylist2 <- mylist[sapply(mylist, function(x) dim(x)[1]) > 0]

This creates the first list with all data.frames that have "df" in their names, while keeping their names as attributes of the list elements.

Then I apply your filter for 1-element data.frames and you just retrieve the names:

names(mylist2)
#### [1] "df1" "df2"
like image 52
agenis Avatar answered Nov 02 '22 00:11

agenis


You need to have a named list. Here is how you could do it:

library(dplyr)
library(purrr)

mylist <- list(df1 = df1, df2 = df2, df3 = df3)

test <- mylist %>%
  map_int(nrow)

test[test > 0] %>% names()
#[1] "df1" "df2"
like image 40
Perry Kuipers Avatar answered Nov 02 '22 00:11

Perry Kuipers