如何在R中创建双参数filter包装函数及解决数据框创建报错？

一、过滤函数实现 #

以下是符合需求的包装函数，能自动识别大小写不同的country列：

filter_by_country <- function(df, country_name) {
  # 统一列名大小写，查找是否存在country相关列
  country_col_idx <- which(tolower(colnames(df)) == "country")
  
  # 没有找到对应列时返回提示
  if (length(country_col_idx) == 0) {
    return("No country data")
  }
  
  # 找到列后，过滤出指定国家的行
  filtered_result <- df[df[[country_col_idx]] == country_name, ]
  return(filtered_result)
}

二、正确创建示例数据框 #

你之前写的df=data.frame(country_name)会报错，原因是没有给数据框传入实际数据，只是把变量名当成了列名。正确写法需要给列指定具体的向量数据：

示例：带country列的数据框 #

# 准备数据向量
country_list <- c("China", "USA", "Japan", "China")
value_list <- c(100, 200, 150, 120)

# 创建数据框
df <- data.frame(country = country_list, value = value_list)

示例：带Country列的数据框 #

如果列名是大写开头的Country，同样可以用上面的函数处理：

df <- data.frame(Country = c("Germany", "France", "Germany"), score = c(85, 90, 88))

三、测试函数效果 #

# 测试存在对应列的场景
filter_by_country(df, "China")

# 测试不存在对应列的场景
df_no_country <- data.frame(name = c("Alice", "Bob"), age = c(25, 30))
filter_by_country(df_no_country, "China")

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