diff --git a/Week03/weighted_ali_badalov.py b/Week03/weighted_ali_badalov.py new file mode 100644 index 0000000..c2e973d --- /dev/null +++ b/Week03/weighted_ali_badalov.py @@ -0,0 +1,10 @@ +import random +def weighted_srs(data, n, weights, with_replacement = False): + if len(data) != len(weights): + raise ValueError + if with_replacement: + return random.choices(data, weights=weights, k=n) + else: + if n > len(data): + raise ValueError + return random.sample(data, k=n)