Изменение длины ветви в дендрограмме (pheatmap) - PullRequest
Новая
       60

Изменение длины ветви в дендрограмме (pheatmap)

0 голосов
/ 13 января 2020

Я пытаюсь построить тепловую карту с библиотечной pheatmap в R. Я думаю , что по умолчанию длина ветви пропорциональна «разнице» кластеров, которые были объединены на этом этапе. Я бы хотел, чтобы это произошло, так что это фиксированное значение, потому что для моей цели это выглядит очень странно!

Если у кого-то есть идея, как я могу это исправить, я был бы очень рад.

Вот пример кода 

library(pheatmap)
test = matrix(rnorm(6000), 100, 60)
pheatmap(test)

Ура!

1 Ответ

2 голосов
/ 16 января 2020

Вот пример двух групп столбцов с высокой разнородностью: 

library(pheatmap)
test = cbind(matrix(rnorm(3000), 100, 30),
            matrix(rnorm(3000)+10, 100, 30))
pheatmap(test)

enter image description here

TIn pheatmap дендрограмма строится с помощью pheatmap:::draw_dendrogram Длина функций и ветвей сохраняется в объекте h.
Ниже я определяю ветви равной длины, добавляя команду
hc$height <- cumsum(rep(1/length(hc$height), length(hc$height)))
следующим образом: 

draw_dendrogram <- function(hc, gaps, horizontal = T) {
    # Define equal-length branches
    hc$height <- cumsum(rep(1/length(hc$height), length(hc$height)))
    h = hc$height/max(hc$height)/1.05
    m = hc$merge
    o = hc$order
    n = length(o)
    m[m > 0] = n + m[m > 0]
    m[m < 0] = abs(m[m < 0])
    dist = matrix(0, nrow = 2 * n - 1, ncol = 2, dimnames = list(NULL, 
        c("x", "y")))
    dist[1:n, 1] = 1/n/2 + (1/n) * (match(1:n, o) - 1)
    for (i in 1:nrow(m)) {
        dist[n + i, 1] = (dist[m[i, 1], 1] + dist[m[i, 2], 1])/2
        dist[n + i, 2] = h[i]
    }
    draw_connection = function(x1, x2, y1, y2, y) {
        res = list(x = c(x1, x1, x2, x2), y = c(y1, y, y, y2))
        return(res)
    }
    x = rep(NA, nrow(m) * 4)
    y = rep(NA, nrow(m) * 4)
    id = rep(1:nrow(m), rep(4, nrow(m)))
    for (i in 1:nrow(m)) {
        c = draw_connection(dist[m[i, 1], 1], dist[m[i, 2], 1], 
            dist[m[i, 1], 2], dist[m[i, 2], 2], h[i])
        k = (i - 1) * 4 + 1
        x[k:(k + 3)] = c$x
        y[k:(k + 3)] = c$y
    }
    x = pheatmap:::find_coordinates(n, gaps, x * n)$coord
    y = unit(y, "npc")
    if (!horizontal) {
        a = x
        x = unit(1, "npc") - y
        y = unit(1, "npc") - a
    }
    res = polylineGrob(x = x, y = y, id = id)
    return(res)
}
# Replace the non-exported function `draw_dendrogram` in `pheatmap`:
assignInNamespace(x="draw_dendrogram", value=draw_dendrogram, ns="pheatmap")

pheatmap(test)

результат:

enter image description here

...