R语言中使用lapply提取分组数据框指定行时出现维度错误求解决

你遇到的incorrect number of dimensions错误，是因为split.data.frame生成的列表里存在非二维data.frame的元素——大概率是某个国家的分组只有1行数据，被默认简化成了一维向量，导致你用二维索引x[rank, 1:2]时出错。

方法1：修复split后的列表元素类型 #

先强制把列表里的所有元素转成data.frame，避免向量干扰：

# 重新处理分组列表，确保每个元素都是data.frame
group_by_country <- lapply(split(ordered_needed_data, ordered_needed_data$Country), as.data.frame)

然后修改辅助函数，同时处理"某个国家行数不足rank"的情况（比如要取第2名，但该国只有1家公司）：

helperfunction <- function(x, rank) {
  if (nrow(x) >= rank) {
    # 用列名代替索引，更安全，避免列顺序变动出问题
    x[rank, c("Country", "Company")]
  } else {
    # 行数不足时返回NA，保证结构统一
    data.frame(Country = x$Country[1], Company = NA)
  }
}

最后应用函数并合并成结果data.frame：

# 取每个国家第2名，替换rank为你需要的数值（比如7）
result_list <- lapply(group_by_country, helperfunction, rank = 2)
final_result <- do.call(rbind, result_list)

方法2：用dplyr更简洁实现（推荐） #

不用手动split，直接用分组操作一步完成，代码更简洁且自动处理异常：

library(dplyr)

# 取每个国家满意度排名第2的公司，替换数字为目标排名
final_result <- ordered_needed_data %>%
  group_by(Country) %>%
  slice(2) %>%
  select(Country, Company) %>%
  ungroup()

如果需要保留行数不足的国家并填充NA，可以改成：

final_result <- ordered_needed_data %>%
  group_by(Country) %>%
  summarise(Company = ifelse(n() >= 2, Company[2], NA))

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