是否可以在`matplotlib_venn` venn图中放置标签而不是设置尺寸?

时间:2018-05-23 09:35:10

标签: python matplotlib matplotlib-venn

例如,以下代码:

import matplotlib.pyplot as plt
import matplotlib_venn as mv
import seaborn

cols = seaborn.color_palette(['orange', 'darkblue'])
plt.figure()

s1 = {'TIMP3', 'COL4A1', 'SERPINE1', 'SPARC', 'FBLN2', 'THBS2', 'SERPINE2', 'COL1A1', 'LTBP2'}
s2 = {'SPARK', 'FBLN1', 'VCAN', 'TIMP1', 'SERPINE2', 'SKIL', 'LTBP2', 'TIMP3', 'THBS2', 'TGFBR1', 'BGN', 'VEGFA', 'COL1A1', 'FBN1', 'FN1', 'COL4A1', 'MMP2', 'MMP14', 'COL1A2', 'PMPEA1', 'FGF2'}

mv.venn2_unweighted([s1, s2], ['Adult', 'Senescent'], set_colors=cols)
plt.show()

生产: enter image description here

但是,如果每个字符串属于每个字符串的标签位于圆圈内(而不是每个集合的大小),那么对我来说会很有用。这可能吗?

1 个答案:

答案 0 :(得分:1)

您可以通过ID访问标签,ID是一个字符串,数字0和1表示集合。例如。如果v.get_label_by_id("01")是使用中的虚拟图,v会为您提供第二个圆圈的标签。

您可能因此将图表下方的标签设置为空字符串,而是为每个内部标签设置新文本,重新使用已存在的尺寸标签。

import matplotlib.pyplot as plt
import matplotlib_venn as mv
import seaborn

cols = seaborn.color_palette(['orange', 'darkblue'])
plt.figure()

s1 = {'TIMP3', 'COL4A1', 'SERPINE1', 'SPARC', 'FBLN2', 'THBS2', 'SERPINE2', 'COL1A1', 'LTBP2'}
s2 = {'SPARK', 'FBLN1', 'VCAN', 'TIMP1', 'SERPINE2', 'SKIL', 'LTBP2', 'TIMP3', 'THBS2', 'TGFBR1', 'BGN', 'VEGFA', 'COL1A1', 'FBN1', 'FN1', 'COL4A1', 'MMP2', 'MMP14', 'COL1A2', 'PMPEA1', 'FGF2'}

v = mv.venn2_unweighted([s1, s2], ["",""], set_colors=cols)

labels = ['Adult', 'Senescent']

def label_by_id(label, ID):
    num = v.get_label_by_id(ID).get_text() 
    v.get_label_by_id(ID).set_text(label+"\n"+num)

for label, ID in zip(labels, ["10", "01"]):
    label_by_id(label, ID)

plt.show()

enter image description here