迹忆客 专注技术分享

本篇文章将讨论如何使用 Python 和 Matplotlib 库生成维恩图。

在 Python 中使用 Matplotlib 创建维恩图

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

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

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

pip install matplotlib-venn

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

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()

输出：

加权维恩图显示正面非常小。如果我们想保持两个维恩图圆的大小相同，我们可以使用 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()

输出：

下一个示例使用 venn2_circles() 函数生成空白的两个圆圈，你无法在其中进一步对此图进行注释。此方法不接受 set_labels 和 set_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()

输出：

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

项目应按以下顺序输入。

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()

输出：

我们还可以使用 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()

输出：

最后，我们可以使用 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()

输出：