ggplot2:热图中的平均行

时间:2013-11-20 15:23:52

标签: r ggplot2 heatmap

假设我使用函数geom_raster()(来自ggplot2)创建了热图。

在表格底部添加一行的智能方法是显示(在我的情况下)'平均回报率'在考虑的时间内每个月?

在1985年至2013年期间和平均行之间留下一些空间会很好,也许是警察的颜色和案例'可以定制。

我的代码的核心如下(对象molten包含我的数据,最初是通过melt()的{​​{1}}函数传递的矩阵。

reshape2

image

2 个答案:

答案 0 :(得分:7)

它不是很简洁,但我认为这应该做你需要的。

你没有提供数据集,所以我做了一些。此外,LtoMMtoH函数未包含在我能找到的任何R包中,因此我进行了快速Google搜索并找到了here

以下代码生成一个带有facets的图hm2,以便在底部生成“Mean Return”行:

require(reshape2)
require(ggplot2)

# Random data
set.seed(100)
casted = data.frame(Var1 = rep(1985:2013, times=12), Var2 = rep(month.abb, each=29), return = rnorm(12*29, 0, 9))
molten = melt(casted, id.vars = c("Var1", "Var2"))

LtoM <-colorRampPalette(c('red', 'yellow' ))
Mid <- "snow3"
MtoH <-colorRampPalette(c('lightgreen', 'darkgreen'))

# Averages
monthly.avg = cbind(Var1 = rep("Mean", 12), dcast(molten, Var2 ~ ., mean))
colnames(monthly.avg)[3] = "Mean"
molten2 = merge(molten, melt(monthly.avg), all.x = TRUE, all.y = TRUE)

# New plot
hm2 = 
  ggplot(data = molten2, aes(x = factor(Var2, levels = month.abb), y=Var1, fill=value)) +
  geom_raster() +
  scale_fill_gradient2(low=LtoM(100), mid=Mid, high=MtoH(100)) +
  labs(fill='% Return') +
  geom_text(aes(label=paste(sprintf("%.1f %%", value))), size = 4) +
  xlab(label = NULL) + ylab(label = NULL) +
  theme_bw() +
  theme(axis.text.x = element_text(size = 10, hjust = 0, vjust = 0.4, angle=90)) +
  facet_grid(variable ~ ., scales = "free_y", space = "free_y") +    # grid layout
  theme(strip.background = element_rect(colour = 'NA', fill = 'NA'), strip.text.y = element_text(colour = 'white'))    # remove facet labels

给出了以下情节:

Plot with Mean Row

答案 1 :(得分:5)

这个怎么样:

我创建了一个网格来模拟你的数据

主要变化,是预先计算聚合和“spacer”数据行,并添加到熔化状态, 然后添加scale_y_discrete,以便标记行, 然后确保格式适用于没有%标签的灰色间隔条(代码中的注释)

如果您在问题中包含数据(或样本)

,将来会更容易

plot

require(ggplot2)

molten<-expand.grid(c("Jan","Feb","Mar","Apr","May","Jun","Jul","Aug","Sep","Oct","Nov","Dec"),1985:2013,0)
colnames(molten)<-c("Var2","Var1","value")
molten$value=(runif(nrow(molten))*60)-30

#create means
means<-aggregate(molten[,c(1,3)], by=list(molten$Var2),FUN=mean, na.rm=TRUE)
colnames(means)<-c("Var2","Var1","value")
means$Var1<-"MEANS"

#create spacer bar
spacer<-means
spacer$Var1<-" "
spacer$value<-NA

#append them to the data
molten<-rbind(molten,spacer,means)


hm <- ggplot(data = molten, aes(x = Var2, y=Var1, fill=value)) + 
geom_raster() +
# replaced your functions for ease of use
scale_fill_gradient2(low="red", mid="yellow", high="green",na.value="grey") +
labs(fill='% Return') +
# don't format the NA vals with %, return blank
geom_text(aes(label=ifelse((is.na(value)),"",paste(sprintf("%.1f %%", value)))), size = 4) +
# make the scale discrete to add labels and enforce order (use a blank space for the spacer)
scale_y_discrete(limits = c("MEANS"," ",1985:2013)) +
xlab(label = NULL) + ylab(label = NULL) +
theme_bw() +
theme(axis.text.x = element_text(size = 10, hjust = 0, vjust = 0.4, angle=90))

hm