如何在R lattice的cloud/wireframe中映射颜色至非x/y/z列？

不管是分类变量还是连续数值列，都可以通过col参数直接指定目标列来实现颜色映射，和y轴变量完全独立，以下是两种常见场景的实现方法：

1. 分类变量的颜色映射 #

如果要映射的是分组类的列（比如物种、类别标签），直接把该列传给col参数，再用auto.key自动生成图例：

library(lattice)

# 以iris数据集为例，将颜色映射到Species分类列
cloud(Petal.Length ~ Sepal.Length * Sepal.Width, 
      data = iris,
      col = iris$Species,  # 这里指定颜色映射到Species列
      pch = 16,  # 设置点的形状
      auto.key = list(space = "right", title = "物种"),  # 右侧显示图例并设置标题
      main = "3D散点图：颜色映射到分类列")

运行后，不同物种的点会显示不同颜色，右侧自动生成对应图例。

2. 连续数值列的颜色映射 #

如果是连续数值列，需要先把数值转换成对应的颜色向量，再传给col参数，同时手动构建颜色图例：

library(lattice)

# 第一步：创建连续颜色调色板
color_palette <- colorRampPalette(c("浅蓝", "草绿", "深红"))(100)
# 第二步：把目标列（比如Petal.Width）的数值映射到调色板的索引
color_index <- findInterval(iris$Petal.Width, 
                           seq(min(iris$Petal.Width), max(iris$Petal.Width), length.out = 101))

# 绘制3D散点图
cloud(Petal.Length ~ Sepal.Length * Sepal.Width, 
      data = iris,
      col = color_palette[color_index],  # 颜色映射到Petal.Width列
      pch = 16,
      main = "3D散点图：颜色映射到连续数值列",
      # 手动设置连续颜色的图例
      key = list(space = "right",
                 rectangles = list(col = color_palette, height = 1, width = 1),
                 text = list(as.character(round(seq(min(iris$Petal.Width), max(iris$Petal.Width), length.out = 5), 2))),
                 title = "花瓣宽度"))

关键提示 #

• cloud函数的语法是z ~ x * y，y只是其中一个坐标轴变量，颜色映射和坐标轴变量完全独立，只要在col参数传入你想要映射的列的向量即可。
• 如果需要更灵活的分组配色，也可以结合groups参数，比如groups = Species, col = c("#FF0000", "#00FF00", "#0000FF")，手动指定每组的颜色。

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