Matthias Munz
Matthias Munz

Reputation: 3765

How to create a heatmap with continuous scale using ggplot2 in R

I have got a data frame with several 1000 rows in the form of

group = c("gr1","gr1","gr1","gr1","gr1","gr1","gr1","gr1","gr1","gr1","gr2","gr2","gr2","gr2","gr2","gr2","gr2","gr2","gr2","gr2","gr3","gr3","gr3","gr3","gr3","gr3","gr3","gr3","gr3","gr3")
pos = c(1,2,3,4,5,6,7,8,9,10,1,2,3,4,5,6,7,8,9,10,1,2,3,4,5,6,7,8,9,10)
color = c(2,2,2,2,3,3,2,2,3,2,1,2,2,2,1,1,1,1,1,1,2,2,2,2,2,2,1,1,2,2)
df = data.frame(group, pos, color)

and would like to make a kind of heatmap in which one axes has a continuous scale (position). The color column is categorical. However due to the large amount of data points I want to use binning, i.e. use it as a continuous variable.

This is more or less how the plot should look like: enter image description here

I can't think of a way to create such a plot using ggplot2/R. I have tried several geometries, e.g. geom_point()

ggplot(data=df, aes(x=strain, y=pos, color=color)) + 
  geom_point() +
  scale_colour_gradientn(colors=c("yellow", "black", "orange"))

enter image description here

Thanks for your help in advance.

Upvotes: 0

Views: 3735

Answers (1)

Mandar
Mandar

Reputation: 1779

Does this help you?

library(ggplot2)
group = c("gr1","gr1","gr1","gr1","gr1","gr1","gr1","gr1","gr1","gr1","gr2","gr2","gr2","gr2","gr2","gr2","gr2","gr2","gr2","gr2","gr3","gr3","gr3","gr3","gr3","gr3","gr3","gr3","gr3","gr3")
pos = c(1,2,3,4,5,6,7,8,9,10,1,2,3,4,5,6,7,8,9,10,1,2,3,4,5,6,7,8,9,10)
color = c(2,2,2,2,3,3,2,2,3,2,1,2,2,2,1,1,1,1,1,1,2,2,2,2,2,2,1,1,2,2)
df = data.frame(group, pos, color)
ggplot(data = df, aes(x = group, y = pos)) + geom_tile(aes(fill = color)) 

Looks like this enter image description here

Improved version with 3 color gradient if you like

library(scales)
ggplot(data = df, aes(x = group, y = pos)) + geom_tile(aes(fill = color))+ scale_fill_gradientn(colours=c("orange","black","yellow"),values=rescale(c(1, 2, 3)),guide="colorbar")

Version 2 with 3  color gradient of your choice

Upvotes: 3

Related Questions