循环返回给定条件R的字符串返回数字

问题描述

我有一个问题。例如，我有一列将地区指示为字符串"district1"。但是，现在我想添加另一列，其中仅包含区号。我正在使用if else语句，但这很累。

下面我仅举一个我拥有的30多个地区的例子：

### generate dataframe
 d <- data.frame(variable=c("district1","district1","district2","district3"))


 d$id1 = ifelse(d$variable=="district1",1,ifelse(d$variable=="district2",2,ifelse(d$variable=="district3",3,NA)))

因此，我想知道是否可以通过循环执行此操作。但是循环结构做得不好：

#  for (i in 1:30){
#    d$id1 = ifelse(d$variable=="districti",i,NA) 
# }

有人可以帮我做一个更优雅的方式吗？太好了 最好的

解决方法

也许您正在寻找这样的解决方案：

library(tydiverse)
d %>% 
  group_by(variable) %>% 
  mutate(id = cur_group_id())

d %>% 
  mutate(id = str_remove_all(variable,pattern = "[a-zA-Z]"))

   variable id
1 district1  1
2 district1  1
3 district1  1
4 district1  1
5 district2  2
6 district2  2
7 district3  3

,

您可以尝试这种方法

library(dplyr)
library(stringr)
d %>% 
 mutate(num = str_extract(variable,"\\d+"))
#   variable num
# 1 district1   1
# 2 district1   1
# 3 district1   1
# 4 district1   1
# 5 district2   2
# 6 district2   2
# 7 district3   3

使用str_extract提取数字； \\d+至少匹配数字一次