代码之家 › 专栏 › 技术社区 › RoyBatty

如何根据字符串向量选择数据帧的列,以确保完全一致?

grep dplyr r

RoyBatty · 技术社区 · 1 年前

我有一个数据框架,其中包含以下列名称:

NewYork_10
NewYork_20
NewYork3_10
NewYork3_20
NewYork4_10
NewYork4_20
HongKong_10
HongKong_20
SanFrancisco_10
SanFrancisco_20

我有一个向量:

list <- c("NewYork", "SanFrancisco")

我想要一个脚本来创建一个新的数据帧,选择那些在下划线之前有完全相同字符串的列。在上面给出的示例中,您将获得一个具有以下列的新数据帧。纽约_10 纽约_20 旧金山_10 SanFrancisco_20

我用grep做了几次尝试:

dplyr::select(matches(list_cities))

dplyr::select(matches(paste0(list_cities), "_"))

甚至对向量使用锚点,我不确定这是否可能。

dplyr::select(matches(paste0("^",list_cities, "_.*")))

但在任何情况下,它都会捕获以给定子字符串开头的向量的所有值。

2 回复 | 直到 1 年前

GuedesBF 1 年前

我们也可以使用 matches

df %>%
    select(matches("(NewYork)|(SanFrancisco)_.*")

GKi 1 年前

您可以尝试:

df[grep("^(NewYork|SanFrancisco)_", names(df))]
#df[grep(paste0("^(", paste0(name_list, collapse="|"), ")_"), names(df))] #Alternative using the name_list
#  NewYork_10 NewYork_20 SanFrancisco_10 SanFrancisco_20
#1          1          1               1               1

或使用 dplyr::select

library(tidyverse)
df %>% select(matches("^(NewYork|SanFrancisco)_"))
#  NewYork_10 NewYork_20 SanFrancisco_10 SanFrancisco_20
#1          1          1               1               1

哪里 ^ 是字符串的开始, (NewYork|SanFrancisco) 比赛 NewYork 或 SanFrancisco 然后 _ .

或者使用 startsWith :

df[Reduce(`|`, lapply(paste0(name_list, "_"), startsWith, x=names(df)))]
#  NewYork_10 NewYork_20 SanFrancisco_10 SanFrancisco_20
#1          1          1               1               1

数据(取自@benson23)

df <- data.frame(NewYork_10 = 1,
           NewYork_20 = 1,
           NewYork3_10 = 1,
           NewYork3_20 = 1,
           NewYork4_10 = 1,
           NewYork4_20 = 1,
           HongKong_10 = 1,
           HongKong_20 = 1,
           SanFrancisco_10 = 1,
           SanFrancisco_20 = 1)

name_list <- c("NewYork", "SanFrancisco")

推荐文章

Marc B. · 使用ggplot2创建条形图时“缺少值”

1 年前

deschen · tidyverse与外部向量发生突变,该外部向量的元素是数据帧中的列值

1 年前

Laura · 在Shiny中使用可排序的包拖放名称,这些名称将成为图表

1 年前

Mallikarjun M · 如何使用随机森林进行时间序列预测?

1 年前

ly li · 模型摘要:当表格形状改变时,拟合优度消失

1 年前

C.Robin · 将marginaffects::predictions()的结果连接回main df?

1 年前

monotonic · 如何将格式为“col1+col3+col4”的数据帧的行名转换为一列数字向量“c(1,3,4)”?

2 年前

Shawn Hemelstrand · 为什么我的自定义errorbar函数不能在R中工作?

2 年前

RoyBatty · 统计每个字符在整个数据集中出现的次数

2 年前

stats_noob · R: 记录某个“行为”发生的循环的索引?

2 年前