In [1]:

    

import matplotlib.pyplot as plt
from itertools import combinations
from simple_venn import venn2, venn3, venn4

%matplotlib inline


    

In [2]:

    

fig, axes = plt.subplots(1, 3, figsize=(24, 8))

# two-way venn
sets = 'A B'.split()
subsets = [''.join(combo) for i in range(1, 3) for combo in combinations(sets, i)]
ax = axes[0]
venn2(subsets, ax=ax)
ax.set_title('venn2', fontsize=24)

# three-way venn
sets = 'A B C'.split()
subsets = [''.join(combo) for i in range(1, 4) for combo in combinations(sets, i)]
ax = axes[1]
venn3(subsets, ax=ax)
ax.set_title('venn3', fontsize=24)

# four-way venn
sets = 'A B C D'.split()
subsets = [''.join(combo) for i in range(1, 5) for combo in combinations(sets, i)]
ax = axes[2]
venn4(subsets, ax=ax)
ax.set_title('venn4', fontsize=24)

fig.suptitle('simple_venn Demo', fontsize=30)
plt.savefig('demo.png', bbox_inches='tight')