使用 ggplot2 自动绘制所有 data.table 列的最佳方法

Question

我正在尝试利用 data.table 和 ggplot2 功能中的高级技巧来创建一个简单而强大的函数，该函数会自动将任意 data.table 的所有列 (y) 绘制（在一个图像中！）为输入列 (x) 的函数，可选地以列 (k) 为条件 - 这样我们就可以使用单行快速可视化所有数据列，如下所示：

dt <- data.table(diamonds[1:100,])[order(carat),cut:=as.character(cut)] 

plotAllXYbyZ(dt)
plotAllXYbyZ(dt, x="carat", k="color")
plotAllXYbyZ(dt, x=1, y=c(2,8:10), k=3)

澄清：挑战在于列可以是任何一种类型（数字、字符或因子）。我们想要一个自动处理这个问题的函数。 - 即它应该能够使用 melt 和 ggplot 绘制所有请求的列，正如我在下面的答案中所尝试的那样。

更新：我的代码发布在下面的答案中。它是功能性的（即显示所需的图）。但是它有一个问题，即：它修改了原始的data.table。 - 为了解决这个问题，我在这里提出了一个新问题：Don't want original data.table to be modified when passed to a function

Answer 1

我希望这对你有用：

plotAllXYbyZ <- function(dt, x, y, z) {
  # to make sure all columns to be melted for ploting are numerical 
  dt[, (y):= lapply(.SD, function(x) {as.numeric(as.character(x))}), .SDcols = y]
  dts <- melt(dt, id = c(x,z), measure = y)
  ggplot(dts, aes_string(x = colnames(dt)[x], y = "value", colours = colnames(dt)[z])) +
    geom_line() + facet_wrap(~ variable)
}

dt <- data.table(mtcars)    

plotAllXYbyZ(dt, x=1, y=3:10, z=2)

Answer 2

感谢上面的 cmets，下面是实现所需输出的代码。 - 下图显示了为这些行生成的输出：

    dtDiamonds <- data.table(diamonds[1:100,])[order(carat),cut:=as.character(cut)]
    plotAllXYbyZ(dtDiamonds);   
    plotAllXYbyZ(dtDiamonds, x="carat", k="color") 
    plotAllXYbyZ(dtDiamonds, x=1, y=c(2,8:10), k=3)

为了做到这一点，我必须引入一个函数来将所有内容转换为数字。 唯一剩下的问题是原来的dtDiamonds 被修改了！ - 因为'：='。为了解决这个问题，我在这里发布了一个单独的问题：为了解决这个问题，我在这里提出了一个新问题：Don't want original data.table to be modified when passed to a function。更新：这个问题现在通过使用<-copy(dt) 而不是<-dt 得到解决。

# A function to convert factors and characters to numeric. 
my.as.numeric <- function (x) {
  if (is.factor(x)) {
    if (T %in% is.na(as.numeric(as.character(x)))) # for factors like "red", "blue"
      return (as.numeric(x))   
    else                                           # for factors like  "20", "30", ...
      return (as.numeric(as.character(x)))         # return: 20, 30, ...
  }
  else if (is.character(x)) {
    if (T %in% is.na(as.numeric(x))) 
      return (as.numeric(as.ordered(x)))  
    else                            # the same: for character variables like "20", "30", ...
      return (as.numeric(x))        # return: 20, 30, ... Otherwise, convert them to factor
    return (x)   
  }
}

 plotAllXYbyZ <- function(.dt, x=NULL, y=NULL, k=NULL) { 
  dt <- copy(.dt)    # NB: If copy is not used, the original data.table will get modified !
  if (is.numeric(x)) x <-  names(dt)[x]
  if (is.numeric(y)) y <-  names(dt)[y]
  if (is.numeric(k)) k <-  names(dt)[k]

  if (is.null(x)) x <- names(dt)[1]    

  "%wo%" <- function(x, y) x[!x %in% y]    
  if (is.null(y)) y <- names(dt) %wo% c(x,k)

  # to make sure all columns to be melted for plotting are numerical 
  dt[, (y):= lapply(.SD, function(x) {my.as.numeric(x)}), .SDcols = y]

  ggplot(melt(dt, id=c(x,k), measure = y)) + 
    geom_step(aes(get(x),value,col=variable))  +
    ifelse (is.null(k), list(NULL), list(facet_wrap(~get(k))) ) + 
    labs(x=x, title=sprintf("variable = F (%s | %s)", x, k))
}

[][]3