R语言无需第三方库如何像命名向量一样为矩阵每个元素命名

原理说明 #

R中的矩阵本质是附加了dim属性的向量，直接用names()赋值的元素名会默认作为属性存储，仅不被默认的print.matrix方法展示，实际上已经可以通过名称直接索引元素：

# 你的原有代码已经实现了元素名绑定
v = 1:10
names(v) = LETTERS[1:10]
m = matrix(v, ncol=2, nrow=5)
# 直接按名称索引即可生效
m["A"] # 输出 1

需要打印时同步展示元素名的两种实现 #

方案1：自定义S3打印方法（推荐，保留矩阵全部运算特性） #

给矩阵新增自定义类，重写对应打印方法，不需要修改原有矩阵逻辑，所有矩阵运算能力都保留：

# 给矩阵添加自定义类标识
class(m) <- c("named_matrix", "matrix")
# 重写自定义类的打印方法
print.named_matrix <- function(x, ...) {
  display_mat <- matrix(
    paste0(names(x), ": ", as.vector(x)),
    nrow = nrow(x),
    ncol = ncol(x)
  )
  dimnames(display_mat) <- dimnames(x)
  print(display_mat, quote = FALSE, ...)
}

调用打印后的效果：

[,1]  [,2] 
[1,] A: 1  F: 6 
[2,] B: 2  G: 7 
[3,] C: 3  H: 8 
[4,] D: 4  I: 9 
[5,] E: 5  J: 10

方案2：元素名映射表（无需修改对象属性） #

如果你不想修改矩阵本身的属性，可以单独维护一张和矩阵同维度的名称映射表，通过自定义函数实现索引：

# 构造和矩阵同维度的名称映射表
elem_names <- matrix(letters[1:10], ncol=2, nrow=5)
m <- matrix(1:10, ncol=2, nrow=5)
# 按名称取值的辅助函数
get_elem <- function(mat, name) {
  mat[elem_names == name]
}
# 调用示例
get_elem(m, "a") # 输出1

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