如何使用R中的分类数据制作3D图？

install.packages("rgl")
library(rgl)

data <- data.frame(
  x = c(1, 2, 3, 4, 5),
  y = c(1, 2, 3, 4, 5),
  z = c(1, 2, 3, 4, 5),
  category = c("A", "B", "A", "C", "B")
)

plot3d(data$x, data$y, data$z, col = c("red", "green", "blue")[as.numeric(data$category)], size = 5)