Можно ли поместить метки, а не задавать размер внутри диаграммы Венна `matplotlib_venn`? - PullRequest
0 голосов
/ 23 мая 2018

Например, следующий код:

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 голосов
/ 23 мая 2018

Вы можете получить доступ к меткам по идентификатору, который представляет собой строку с номерами 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

...