weighted_riza_arslan.py

Week02/weighted_riza_arslan.py

@@ -0,0 +1,8 @@
+import random
+def weighted_srs(data, n, weights, with_replacement=False):
+    w = weights if weights else [1]*len(data)
+    if with_replacement or weights:
+        return random.choices(data, weights=w, k=n)
+    res, d_c = [], list(data)
+    for _ in range(n): res.append(d_c.pop(random.randrange(len(d_c))))
+    return res