summaryrefslogtreecommitdiffstats
path: root/library/python/reservoir_sampling/__init__.py
diff options
context:
space:
mode:
authoryazevnul <[email protected]>2022-02-10 16:46:48 +0300
committerDaniil Cherednik <[email protected]>2022-02-10 16:46:48 +0300
commit9abfb1a53b7f7b791444d1378e645d8fad9b06ed (patch)
tree49e222ea1c5804306084bb3ae065bb702625360f /library/python/reservoir_sampling/__init__.py
parent8cbc307de0221f84c80c42dcbe07d40727537e2c (diff)
Restoring authorship annotation for <[email protected]>. Commit 2 of 2.
Diffstat (limited to 'library/python/reservoir_sampling/__init__.py')
-rw-r--r--library/python/reservoir_sampling/__init__.py32
1 files changed, 16 insertions, 16 deletions
diff --git a/library/python/reservoir_sampling/__init__.py b/library/python/reservoir_sampling/__init__.py
index 7fc579feb43..4ee46ee5e11 100644
--- a/library/python/reservoir_sampling/__init__.py
+++ b/library/python/reservoir_sampling/__init__.py
@@ -1,16 +1,16 @@
-import random
-
-
-def reservoir_sampling(data, nsamples, prng=None):
- if prng is None:
- prng = random
-
- result = []
- for i, entry in enumerate(data):
- if i < nsamples:
- result.append(entry)
- else:
- j = prng.randint(0, i)
- if j < nsamples:
- result[j] = entry
- return result
+import random
+
+
+def reservoir_sampling(data, nsamples, prng=None):
+ if prng is None:
+ prng = random
+
+ result = []
+ for i, entry in enumerate(data):
+ if i < nsamples:
+ result.append(entry)
+ else:
+ j = prng.randint(0, i)
+ if j < nsamples:
+ result[j] = entry
+ return result