check random subset size distributions

distributions.py

from collections import defaultdict
from functools import partial
from json import dumps
from random import *

l = [5, 1, 2, 7, 4]
 
f = lambda x: [i for _, i in sample(list(enumerate(x)),k=randrange(len(x))+1)]

def g(a):
 n=len(a);s=randint(1,2**n-1);t=[j for j in range(n)if s&2**j]
 return [a[i] for i in t]


def test(func):

    probs = defaultdict(int)

    for _ in range(1_000_000):
        s = func(l)
        probs[frozenset(s)] += 1

    return probs
    
    
def main():

    results = {}
    
    for func in [f, g]:
        results[func.__name__] = {str(key): value for key, value in test(func).items()}

    print(dumps(results, indent=2))
    
    
if __name__ == '__main__':
    main()