如何在R语言image()函数中以数值为索引从颜色向量取色绘图

核心逻辑很直接：image()函数会自动将输入矩阵的数值作为颜色向量的索引来选取对应颜色，只需指定col参数为你的颜色向量即可。需要注意的是image()默认会把矩阵的第一行绘制在图像底部，可通过调整坐标轴来匹配直观的矩阵行列顺序。

示例代码 #

# 构造目标矩阵
mat <- matrix(c(1,2,3,3,2,1), nrow = 2, byrow = TRUE)
# 定义对应颜色向量
color_vec <- c("red", "green", "yellow")

# 绘制矩形图
image(mat, col = color_vec, axes = FALSE)
# 添加自定义坐标轴，让矩阵行顺序和图像显示一致
axis(1, at = seq(0, 1, length.out = ncol(mat)), labels = 1:ncol(mat))
axis(2, at = seq(0, 1, length.out = nrow(mat)), labels = nrow(mat):1)
# 添加边框
box()

代码说明 #

• image(mat, col = color_vec)：矩阵中的1、2、3会直接对应color_vec的第1、2、3位颜色，完全实现你需要的颜色映射效果。
• axes = FALSE：关闭默认的数值坐标轴，避免与矩阵行列索引混淆。
• 自定义坐标轴的y轴标签用nrow(mat):1反转，是为了让矩阵的第一行显示在图像顶部，和常规的矩阵阅读顺序保持一致。

内容的提问来源于stack exchange，提问作者G.A.7