Matplotlib 维恩图

Salman Mehmood 2024年2月15日
Matplotlib 维恩图

本教程将讨论如何使用 Python 和 Matplotlib 库生成维恩图。

在 Python 中使用 Matplotlib 创建维恩图

Matplotlib 库使你能够使用某些特定函数生成两个圆和三圆维恩图。使用特定命令,我们可以为维恩图生成加权、未加权和可调的圆。

可调圆圈允许你进一步注释或添加其他信息到你的维恩图。

我们要解决的第一个问题是两圈维恩图,所以让我们开始吧。如果你的电脑上没有安装 matplotlib-venn,请使用以下命令安装它。

pip install matplotlib-venn

然后我们导入我们需要的库。让我们创建一个名为 itemslabels 的列表,其中包含两个标签。

import matplotlib.pyplot as plot
from matplotlib_venn import venn2, venn2_circles, venn2_unweighted

items = [80, 43, 6]
labels = ["set1", "set2"]

venn2() 函数是生成圆圈的加权图,其大小取决于我们在此项目列表中给出的值。这个函数有几个参数。

第一个参数是 subsets,它接受我们在上面创建的传递的列表。第二个参数是 set_labels,它接受标签列表并将其传递给 labels 列表。

第三个参数是 set_colors,在这里我们可以选择为维恩图添加颜色,颜色值取决于标签,所以保持与标签列表相同的数量。

我们还可以使用 alpha 参数设置透明度。这个参数范围是 0 到 1,我们可以传递 0 到 1 之间的浮点值。

venn2(subsets=items, set_labels=labels, set_colors=("red", "green"), alpha=0.7)

完整代码:

import matplotlib.pyplot as plot
from matplotlib_venn import venn2, venn2_circles, venn2_unweighted

items = [80, 43, 6]
labels = ["set1", "set2"]
venn2(subsets=items, set_labels=labels, set_colors=("red", "green"), alpha=0.7)
plot.title("2 weighted venn diagram")
plot.show()

输出:

2 Matplotlib 中的加权维恩图

加权维恩图显示正面非常小。如果我们想保持两个维恩图圆的大小相同,我们可以使用 venn2_unweighted() 函数,它接受与 venn2() 函数相同的参数。

代码:

import matplotlib.pyplot as plot
from matplotlib_venn import venn2, venn2_circles, venn2_unweighted

items = [80, 43, 6]
labels = ["set1", "set2"]
venn2_unweighted(
    subsets=items, set_labels=labels, set_colors=("red", "green"), alpha=0.5
)
plot.title("2 unweighted venn diagram")
plot.show()

输出:

2 Matplotlib 中的未加权维恩图

下一个示例使用 venn2_circles() 函数生成空白的两个圆圈,你无法在其中进一步对此图进行注释。此方法不接受 set_labelsset_colors 参数。

代码:

import matplotlib.pyplot as plot
from matplotlib_venn import venn2, venn2_circles, venn2_unweighted

items = [80, 43, 6]
labels = ["set1", "set2"]
venn2_circles(subsets=items)
plot.title("2 circles venn diagram")
plot.show()

输出:

Matplotlib 中的 2 个圆圈维恩图

现在让我们继续讨论三圈维恩图。三圈维恩图需要三组。

项目应按以下顺序输入。

items=[a,b,ab,c,ac,bc,abc]

让我们为三个集合和标签创建一个新的项目列表。

items = [80, 43, 6, 62, 16, 3, 18]
labels = ["set1", "set2", "set3"]

在这三个维恩图中,我们可以观察到相交部分。

代码:

import matplotlib.pyplot as plot
from matplotlib_venn import venn3, venn3_unweighted

items = [80, 43, 6, 62, 16, 3, 18]
labels = ["set1", "set2", "set3"]

venn3(subsets=items, set_labels=labels, alpha=0.6)
plot.title("3 weighted venn diagram")
plot.show()

输出:

3 Matplotlib 中的加权维恩图

我们还可以使用 venn3_unweighted() 函数创建三个未加权的维恩图,如下例所示。

代码:

import matplotlib.pyplot as plot
from matplotlib_venn import venn3, venn3_unweighted

items = [80, 43, 6, 62, 16, 3, 18]
labels = ["set1", "set2", "set3"]

venn3_unweighted(subsets=items, set_labels=labels, alpha=0.6)
plot.title("3 unweighted venn diagram")

plot.show()

输出:

3 Matplotlib 中的未加权维恩图

最后,我们可以使用 venn3_circles 函数创建一个三圆图。

代码:

import matplotlib.pyplot as plot
from matplotlib_venn import venn3_circles

items = [80, 43, 6, 62, 16, 3, 18]
labels = ["set1", "set2", "set3"]

venn3_circles(subsets=items, linestyle="dashed")
plot.title("3 circle venn diagram")

plot.show()

输出:

Matplotlib 中的 3 个圆圈维恩图

作者: Salman Mehmood
Salman Mehmood avatar Salman Mehmood avatar

Hello! I am Salman Bin Mehmood(Baum), a software developer and I help organizations, address complex problems. My expertise lies within back-end, data science and machine learning. I am a lifelong learner, currently working on metaverse, and enrolled in a course building an AI application with python. I love solving problems and developing bug-free software for people. I write content related to python and hot Technologies.

LinkedIn

相关文章 - Matplotlib Diagram