在热图中标记组

Question

我有一个数组，其中 ij 条目是 i 和 j 区域共有的基因数，它们在< em> i 关于 j 。

标记每个xtick和ytick将使图形过于拥挤。相近 this question和this question我想在x轴上对标签进行分组。

来自Hawrylycz等人（2012）的下图中热图的xticklabels是我想要的一个很好的例子。xticklabels指的是更一般的区域。例如，额叶下的所有柱都与额叶内大脑中的结构相对应。

我不是要复制yticklabels或条形图插图。

我的方法

对于热图中的每个框，我都有一个本体。我选择在一些地区绘制结构，例如只有“额叶和顶叶”。

使用本体我可以发现每个结构的列组的开始和结束索引。 如何使用这些索引绘制组标签？

Answer 1

像这样：

import pandas as pd
from numpy.random import random_integers
from numpy import reshape
import matplotlib.pyplot as plt
from matplotlib.ticker import FixedLocator, FixedFormatter
alph = 'ABCDEFGHIJKLMNOPQRSTUVWXYZ'
lalph = list(alph.lower())
alph = list(alph)

df = pd.DataFrame(random_integers(0,100,(26,26)),columns=alph,
                  index=lalph)

# Two lines just to make a plaid image in imshow 
differ = reshape([sum(df[col2]-df[col]) for col2 in df for col in df], (26,26))
differ = pd.DataFrame(differ, columns=alph,index=alph)

# pick the labels you want
ticks = [2, 14, 18, 19, 22] # C=2 because A=0 because Python is 0-indexed
ticklabels = [alph[x] for x in ticks]

fig = plt.figure(figsize=(3,5))
ax = fig.add_subplot(111)
ax.imshow(differ)
ax.autoscale(False)

# display only the chosen ticks and ticklabels
ax.xaxis.set_major_locator(FixedLocator(ticks))
ax.xaxis.set_major_formatter(FixedFormatter(ticklabels))

你将有一个命名基因的字符串列表，而不是用作字母列表的字符串，但是imshow轴索引仍然是底层numpy数组的索引。