Refactor weighted_srs function

Refactor weighted_srs function for improved readability.

Author: Ebrukoksal

Week02/weighted_ebru_koksal.py

@@ -1,10 +1,10 @@
 import random
 
-def weighted_srs(data, n, weights=None, with_replacement=False):
-  if with_replacement: return random.choices(data, weights=weights, k=n)
-  result, pool, w = [], list(data), (list(weights) if weights else None)
+def weighted_srs(data,n,weights,with_replacement=False):
+  if with_replacement:return random.choices(data,weights=weights,k=n)
+  pool=list(data);w=list(weights) if weights else None;res=[]
   for _ in range(n):
-      i = random.choices(range(len(pool)), weights=w, k=1)[0] if w else random.randrange(len(pool))
-      result.append(pool.pop(i))
-      if w: w.pop(i)
-  return result
+    i=random.choices(range(len(pool)),weights=w,k=1)[0] if w else random.randrange(len(pool))
+    res.append(pool.pop(i))
+    if w:w.pop(i)
+return res