grepl и подмножество из списка векторов? - PullRequest
Новая
       12

grepl и подмножество из списка векторов?

0 голосов
/ 07 ноября 2019

У меня есть список, сгенерированный из приведенного ниже кода, 

df <- USArrests
df <- na.omit(df)
df <- scale(df)
d <- dist(df, method = "euclidean")

# Hierarchical clustering using Complete Linkage
hc1 <- hclust(d, method = "complete" )

library(dendextend)
dend15 <- d %>% hclust(method = "average") %>% as.dendrogram
dend15 %>% plot

subtrees <- partition_leaves(dend15)

Что я хотел бы сделать, это создать новый список с grep для ключевого слова "Maine" . Возможно ли это?

Пример данных: 

head ( subtrees, 20 )

[[1]]
 [1] "North Dakota"   "Maine"          "Iowa"           "New Hampshire"  "Vermont"       
 [6] "South Dakota"   "West Virginia"  "Delaware"       "Rhode Island"   "Massachusetts" 
[11] "New Jersey"     "Arkansas"       "Kentucky"       "Connecticut"    "Minnesota"     
[16] "Wisconsin"      "Idaho"          "Montana"        "Nebraska"       "Wyoming"       
[21] "Virginia"       "Oklahoma"       "Indiana"        "Kansas"         "Ohio"          
[26] "Pennsylvania"   "Hawaii"         "Utah"           "Oregon"         "Washington"    
[31] "Alaska"         "Georgia"        "Tennessee"      "Alabama"        "Louisiana"     
[36] "North Carolina" "Mississippi"    "South Carolina" "California"     "Nevada"        
[41] "Florida"        "Colorado"       "Missouri"       "Texas"          "Illinois"      
[46] "New York"       "Arizona"        "Michigan"       "Maryland"       "New Mexico"    

[[2]]
 [1] "North Dakota"  "Maine"         "Iowa"          "New Hampshire" "Vermont"      
 [6] "South Dakota"  "West Virginia" "Delaware"      "Rhode Island"  "Massachusetts"
[11] "New Jersey"    "Arkansas"      "Kentucky"      "Connecticut"   "Minnesota"    
[16] "Wisconsin"     "Idaho"         "Montana"       "Nebraska"      "Wyoming"      
[21] "Virginia"      "Oklahoma"      "Indiana"       "Kansas"        "Ohio"         
[26] "Pennsylvania"  "Hawaii"        "Utah"          "Oregon"        "Washington"   

[[3]]
[1] "North Dakota"  "Maine"         "Iowa"          "New Hampshire" "Vermont"      
[6] "South Dakota"  "West Virginia"

[[4]]
[1] "North Dakota"  "Maine"         "Iowa"          "New Hampshire"

[[5]]
[1] "North Dakota"

[[6]]
[1] "Maine"         "Iowa"          "New Hampshire"

[[7]]
[1] "Maine"

[[8]]
[1] "Iowa"          "New Hampshire"

[[9]]
[1] "Iowa"

1 Ответ

1 голос
/ 07 ноября 2019

lapply над списком и используйте grep

lapply(subtrees, grep, pattern = "Maine", value = TRUE)

Возможно, вы захотите удалить из него пустые списки, что можно сделать с помощью Filter

Filter(function(x) length(x) > 0, lapply(subtrees, grep, pattern = "Maine", value = TRUE))

#[[1]]
#[1] "Maine"

#[[2]]
#[1] "Maine"

#[[3]]
#[1] "Maine"

#[[4]]
#[1] "Maine"

#[[5]]
#[1] "Maine"

#[[6]]
#[1] "Maine"

tidyverse путь может быть 

purrr::map(subtrees, ~stringr::str_subset(.x, "Maine"))

Чтобы получить индекс списка, который соответствует, мы можем использовать grepl вместе с which

which(sapply(subtrees, function(x) any(grepl("Maine", x))))
#[1] 1 2 3 4 6 7
...