在R语言中基于字符串结尾创建新列的实现方法

嘿，这个需求很常见，用R的字符串处理工具就能轻松搞定！我给你两种实用的方案，一种用基础R（不用额外装包），另一种用更简洁的tidyverse工具，你看哪种顺手：

方法1：基础R实现（无需额外包） #

用基础R的grepl()函数配合正则表达式就能匹配字符串结尾的规则：

# 假设你的数据集名为df
df$Trial <- ifelse(grepl("2$", df$Trtmt), 2, 1)

代码解释： #

• grepl("2$", df$Trtmt)：返回一个逻辑向量，$是正则表达式里的“结尾匹配符”，所以这个表达式会检查Trtmt列的每个字符串是否以"2"结尾，是则返回TRUE，否则FALSE。
• ifelse()：根据前面的逻辑结果赋值——TRUE时给2，其余情况（包括不以数字结尾的字符串）都给1，刚好符合你的需求。

如果需要更严谨地先区分“不以数字结尾”的情况，可以这样写（不过你的需求里非2结尾都给1，所以上面的写法足够）：

df$Trial <- ifelse(grepl("2$", df$Trtmt), 2, 
                   ifelse(!grepl("[0-9]$", df$Trtmt), 1, NA))

方法2：用stringr包（tidyverse风格，可读性更强） #

如果你平时习惯用tidyverse工具链，stringr包的str_ends()函数专门用来检查字符串结尾，代码更直观：

# 先安装并加载stringr（首次使用需要安装）
# install.packages("stringr")
library(stringr)

df$Trial <- ifelse(str_ends(df$Trtmt, "2"), 2, 1)

str_ends(df$Trtmt, "2")直接判断每个字符串是否以"2"结尾，逻辑和基础R的方法一致，但代码更易懂。

示例验证 #

我们用你提到的样例数据测试一下：

# 创建样例数据框
df <- data.frame(
  Plant = c("P1", "P2", "P3", "P4"),
  Trtmt = c("A3-H2", "A9-H", "A3-H", "A9-H2")
)

# 运行方法1的代码
df$Trial <- ifelse(grepl("2$", df$Trtmt), 2, 1)

# 查看结果
print(df)

输出结果：

Plant Trtmt Trial
1    P1 A3-H2     2
2    P2  A9-H     1
3    P3  A3-H     1
4    P4 A9-H2     2

完全符合你的预期！

内容的提问来源于stack exchange，提问作者TheSciGuy