R语言循环打印以B开头的国家名称时遇条件长度大于1错误

问题场景 #

我有一个包含字符串类型country列的dataframe（df），数据如下：

code     country
DK       Denmark
DE       Germany
BEB      Belgium
B        Brazil

尝试用for循环打印以B开头的国家名称，编写的代码如下：

for (i in unique(df$country)) {
  if (i == grep("^B", unique(df$country), value = TRUE)) {
    print(i)
  }
}

运行后出现错误：

Error in if (i == grep("^B", unique(df$country), value = TRUE)) { : 
  the condition has length > 1

错误原因 #

grep("^B", unique(df$country), value = TRUE)返回的是长度为2的向量（c("Belgium", "Brazil")），当你用单个元素i和这个向量做==比较时，会得到一个长度为2的逻辑向量，但if语句仅接受长度为1的判断条件，因此触发报错。

修正方案 #

1. 修正原循环逻辑

直接在循环中对每个i单独判断是否以B开头，用grepl()函数（返回单个逻辑值，符合if的要求）：

for (i in unique(df$country)) {
  if (grepl("^B", i)) {
    print(i)
  }
}

2. 更高效的R向量式写法

R是向量型语言，无需用for循环，直接筛选后打印更简洁高效：

# 筛选并打印
b_countries <- grep("^B", df$country, value = TRUE)
print(b_countries)

# 或一行完成
print(grep("^B", df$country, value = TRUE))

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