def venn(vennset=(1,1,1,1,1,1,1), venncolor=('#00909e', '#f67280', '#ff971d'), vennalpha=0.5, vennlabel=('A', 'B', 'C')): fig = plt.figure() if len(vennset) == 7: venn3(subsets=vennset, set_labels=vennlabel, set_colors=venncolor, alpha=vennalpha) plt.savefig('venn3.png', format='png', bbox_inches='tight', dpi=300) elif len(vennset) == 3: venn2(subsets=vennset, set_labels=vennlabel, set_colors=venncolor, alpha=vennalpha) plt.savefig('venn2.png', format='png', bbox_inches='tight', dpi=300) else: print("Error: check the set dataset")