gpt4 book ai didi

r - 当列名称-值对存储在列表中时过滤数据框?

转载 作者:行者123 更新时间:2023-12-05 09:05:07 24 4
gpt4 key购买 nike

我有一个像这样的数据框:

df <- tibble::rownames_to_column(USArrests, "State") %>% 
tidyr::pivot_longer(cols = -State)

head(df)
# A tibble: 6 x 3
State name value
<chr> <chr> <dbl>
1 Alabama Murder 13.2
2 Alabama Assault 236
3 Alabama UrbanPop 58
4 Alabama Rape 21.2
5 Alaska Murder 10
6 Alaska Assault 263

在一个单独的列表对象 l 中,我有一些列,我需要从数据框中删除这些列。元素名称是列名称,值对应于我要删除的行:

l <- list(State = c("Alabama", "Pennsylvania", "Texas"),
name = c("Murder", "Assault"))

硬编码它会这样做:

dplyr::filter(df, !State %in% c("Alabama", "Pennsylvania", "Texas"), !name %in% c("Murder", "Assault"))

State name value
<chr> <chr> <dbl>
1 Alaska UrbanPop 48
2 Alaska Rape 44.5
3 Arizona UrbanPop 80
4 Arizona Rape 31
5 Arkansas UrbanPop 50
6 Arkansas Rape 19.5
7 California UrbanPop 91
8 California Rape 40.6
9 Colorado UrbanPop 78
10 Colorado Rape 38.7
# ... with 84 more rows

但是,l 经常变化,所以我不能/不想硬编码。我尝试了以下操作,但只评估了最后一个表达式:

library(purrr)
filter_expr <- imap_chr(l, ~ paste0("! ",
.y,
" %in% c(\"",
paste(.x, collapse = "\",\""),
"\")")) %>% parse(text = .)

filter(df, eval(filter_expr))

State name value
<chr> <chr> <dbl>
1 Alabama UrbanPop 58
2 Alabama Rape 21.2
3 Alaska UrbanPop 48
4 Alaska Rape 44.5
5 Arizona UrbanPop 80
6 Arizona Rape 31
7 Arkansas UrbanPop 50
8 Arkansas Rape 19.5
9 California UrbanPop 91
10 California Rape 40.6
# ... with 90 more rows

当过滤条件存储在像 l 这样更适合 tidyverse 的结构中时,是否有一种方法可以过滤 df

我考虑过这个SO answer ,但是,表达式不是动态的。

最佳答案

我们可以在 filter 中使用 across 循环遍历 'l' 的 names,通过对 'l' 进行子集化来创建逻辑表达式来自列名称的键 (cur_column()) 和取反 (!)。请注意,cur_column() 目前仅适用于 across 而不是 if_all/if_any (dplyr -1.0 .6R 4.1.0)

library(dplyr)
df %>%
filter(across(all_of(names(l)), ~ !. %in% l[[cur_column()]]))

-输出

# A tibble: 94 x 3
# State name value
# <chr> <chr> <dbl>
# 1 Alaska UrbanPop 48
# 2 Alaska Rape 44.5
# 3 Arizona UrbanPop 80
# 4 Arizona Rape 31
# 5 Arkansas UrbanPop 50
# 6 Arkansas Rape 19.5
# 7 California UrbanPop 91
# 8 California Rape 40.6
# 9 Colorado UrbanPop 78
#10 Colorado Rape 38.7
# … with 84 more rows

如果我们可以设置一个属性,我们可以使用if_all

library(magrittr)
df %>%
mutate(across(all_of(names(l)), ~ set_attr(., 'cn', cur_column()))) %>%
filter(if_all(all_of(names(l)), ~ ! . %in% l[[attr(., 'cn')]]))

或者用imap/reduce

library(purrr)
df %>%
filter(imap(l, ~ !cur_data()[[.y]] %in% .x) %>%
reduce(`&`))

或者另一种选择是anti_join

for(nm in names(l)) df <- anti_join(df, tibble(!! nm := l[[nm]]))

关于r - 当列名称-值对存储在列表中时过滤数据框?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/67694711/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com