r - pheatmap:不适用的颜色

使用R包的Pheatmap绘制热图。有没有办法为输入矩阵中的NA分配颜色？默认情况下，似乎NA的颜色为白色。
例如。:

library(pheatmap)
m<- matrix(c(1:100), nrow= 10)
m[1,1]<- NA
m[10,10]<- NA
pheatmap(m, cluster_rows=FALSE, cluster_cols=FALSE)

谢谢

最佳答案

可能，但需要进行一些黑客操作。

首先，让我们看看pheatmap如何绘制热图。您可以通过在控制台中键入pheatmap并滚动显示输出，或使用edit(pheatmap)来进行检查。

您会发现颜色是使用

mat = scale_colours(mat, col = color, breaks = breaks)

scale_colours函数似乎是pheatmap包的内部函数，但是我们可以使用来检查源代码

getAnywhere(scale_colours)

这使

function (mat, col = rainbow(10), breaks = NA)
{
    mat = as.matrix(mat)
    return(matrix(scale_vec_colours(as.vector(mat), col = col,
        breaks = breaks), nrow(mat), ncol(mat), dimnames = list(rownames(mat),
        colnames(mat))))
}

现在我们需要检查scale_vec_colours，结果是:

function (x, col = rainbow(10), breaks = NA)
{
    return(col[as.numeric(cut(x, breaks = breaks, include.lowest = T))])
}

因此，从本质上讲，pheatmap使用cut决定要使用的颜色。

让我们尝试看看如果周围有NA，cut会做什么:

as.numeric(cut(c(1:100, NA, NA), seq(0, 100, 10)))
  [1]  1  1  1  1  1  1  1  1  1  1  2  2  2  2  2  2  2  2  2  2  3  3  3  3  3  3  3  3
 [29]  3  3  4  4  4  4  4  4  4  4  4  4  5  5  5  5  5  5  5  5  5  5  6  6  6  6  6  6
 [57]  6  6  6  6  7  7  7  7  7  7  7  7  7  7  8  8  8  8  8  8  8  8  8  8  9  9  9  9
 [85]  9  9  9  9  9  9 10 10 10 10 10 10 10 10 10 10 NA NA

返回NA!所以，这是您的问题!

现在，我们如何解决它？
最简单的方法是让pheatmap绘制热图，然后根据需要覆盖NA值。

再次查看pheatmap函数，您将看到它使用grid软件包进行绘图(另请参见此问题:R - How do I add lines and text to pheatmap?)

因此，您可以使用grid.rect将矩形添加到NA位置。
我要做的是通过反复试验找到热图边界的坐标，然后从那里开始绘制矩形。

例如:

library(pheatmap)
m<- matrix(c(1:100), nrow= 10)
m[1,1]<- NA
m[10,10]<- NA

hmap <- pheatmap(m, cluster_rows=FALSE, cluster_cols=FALSE)
# These values were found by trial and error
# They WILL be different on your system and will vary when you change
# the size of the output, you may want to take that into account.
min.x <- 0.005
min.y <- 0.01
max.x <- 0.968
max.y <- 0.990
width <- 0.095
height <- 0.095

coord.x <- seq(min.x, max.x-width, length.out=ncol(m))
coord.y <- seq(max.y-height, min.y, length.out=nrow(m))

for (x in seq_along(coord.x))
  {
  for (y in seq_along(coord.y))
    {
    if (is.na(m[x,y]))
        grid.rect(coord.x[x], coord.y[y], just=c("left", "bottom"),
                  width, height, gp = gpar(fill = "green"))
    }
  }

更好的解决方案是使用pheatmap函数破解edit的代码，并按您的意愿处理NA。

关于r - pheatmap:不适用的颜色，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/25929991/