如何用R函数确定矩阵中向量所在的行（附示例）

Hey，刚好遇到过类似的需求，这就给你几个靠谱的解决办法，完美匹配你要的结果！

首先先把示例数据定义好：

# 定义题目中的两个矩阵
a1 <- rbind(c(1,3), c(1,2))
A1 <- rbind(c(1, 2), c(1, 4), c(1,3), c(2, 4))

方法1：基础R字符串匹配法 #

这是最直接的基础R实现，把每一行拼接成字符串后用match函数找位置：

# 将每行转为逗号分隔的字符串，再匹配行号
apply(a1, 1, function(x) {
  match(paste(x, collapse = ","), apply(A1, 1, paste, collapse = ","))
})

运行后会返回[1] 3 1，完全符合你的预期。

方法2：dplyr数据框连接法 #

如果你习惯用tidyverse工具链，用dplyr的连接操作会更直观：

# 先加载dplyr包（如果没安装先运行install.packages("dplyr")）
library(dplyr)

# 把矩阵转成数据框并添加行号
a1_with_row <- as.data.frame(a1) %>% mutate(row_idx_a = row_number())
A1_with_row <- as.data.frame(A1) %>% mutate(row_idx_A = row_number())

# 内连接后提取A1的行号
inner_join(a1_with_row, A1_with_row, by = c("V1", "V2")) %>% 
  pull(row_idx_A)

这个方法的好处是可读性强，处理复杂矩阵时也不容易出错，结果同样是[1] 3 1。

方法3：逐行完全匹配法 #

如果不想用字符串拼接，也可以直接逐行比较元素是否完全相等：

# 对a1的每一行，找到A1中完全匹配的行号
apply(a1, 1, function(x) {
  which(apply(A1, 1, function(y) all(y == x)))
})

这个方法更“硬核”，直接比较数值，适合对精度要求高的场景。

小提示：如果A1中有重复行，match只会返回第一个匹配的行号，而which会返回所有匹配的行号，你可以根据实际需求选择合适的方法~

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