def plot_ven(labels, names):
if len(names) == 2:
return pv.venn2(labels, names=names)
elif len(names) == 3:
return pv.venn3(labels, names=names)
elif len(names) == 4:
return pv.venn4(labels, names=names)
elif len(names) == 5:
return pv.venn5(labels, names=names)
elif len(names) == 6:
return pv.venn6(labels, names=names)
else:
raise Error("Unsupported Venn")
fill=['number', 'logic'])
fig, ax = venn.venn4(labels, names=['list 1', 'list 2', 'list 3', 'list 4'])
fig.savefig('venn4.png', bbox_inches='tight')
plt.close()
labels = venn.get_labels(
[range(10),
range(5, 15),
range(3, 8),
range(8, 17),
range(10, 20)],
fill=['number', 'logic'])
fig, ax = venn.venn5(labels,
names=['list 1', 'list 2', 'list 3', 'list 4', 'list 5'])
fig.savefig('venn5.png', bbox_inches='tight')
plt.close()
labels = venn.get_labels([
range(10),
range(5, 15),
range(3, 8),
range(8, 17),
range(10, 20),
range(13, 25)
],
fill=['number', 'logic'])
fig, ax = venn.venn6(
labels, names=['list 1', 'list 2', 'list 3', 'list 4', 'list 5', 'list 6'])
fig.savefig('venn6.png', bbox_inches='tight')
plt.close()