python--Venn图及upsetplot进阶

最新推荐文章于 2024-12-14 14:42:35 发布

晏九

最新推荐文章于 2024-12-14 14:42:35 发布

阅读量3.7k

点赞数 6

分类专栏： python绘图

本文链接：https://blog.csdn.net/weixin_42670653/article/details/116460184

版权

python绘图专栏收录该内容

10 篇文章

订阅专栏

本文展示了如何利用Venn图和UpsetPlot来直观展示多个集合的交集情况。通过Python的`venn`库创建了3个集合的Venn图，并进一步使用`matplotlib`和`upsetplot`库展示了2到6个集合的复杂交集，以及如何自定义颜色映射和显示计数。UpsetPlot作为一种更有效的工具，能清晰地揭示多个集合之间的交互关系，其API可用于创建交互式的集合分析图表。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

简介

Venn图用来看几个集合相交的情况，通常下，2-4个集合看Venn图是比较直观的，但是集合数目再多就不适合了，需要引入另外一种图。

venn

3个


from venn import venn

musicians = {
    "Members of The Beatles": {"Paul McCartney", "John Lennon", "George Harrison", "Ringo Starr"},
    "Guitarists": {"John Lennon", "George Harrison", "Jimi Hendrix", "Eric Clapton", "Carlos Santana"},
    "Played at Woodstock": {"Jimi Hendrix", "Carlos Santana", "Keith Moon"}
}
venn(musicians)

2-6个

from matplotlib.pyplot import subplots
from itertools import chain, islice
from string import ascii_uppercase
from numpy.random import choice

_, top_axs = subplots(ncols=3, nrows=1, figsize=(18, 5))
_, bot_axs = subplots(ncols=2, nrows=1, figsize=(18, 8))
cmaps = ["cool", list("rgb"), "plasma", "viridis", "Set1"]
letters = iter(ascii_uppercase)

for n_sets, cmap, ax in zip(range(2, 7), cmaps, chain(top_axs, bot_axs)):
    dataset_dict = {
        name: set(choice(1000, 700, replace=False))
        for name in islice(letters, n_sets)
    }
    venn(dataset_dict, fmt="{percentage:.1f}%", cmap=cmap, fontsize=8, legend_loc="upper left", ax=ax)

upsetplot

可以更直观的看出几个set的的交集情况

api

https://upsetplot.readthedocs.io/en/stable/api.html

from upsetplot import generate_counts
example = generate_counts()
example

这个数据可以通过对dataframe groupby获得（类似数据透视表）

from upsetplot import plot
plot(example)

from upsetplot import plot
plot(example,show_counts='%d')