diff options
Diffstat (limited to 'lib/utils/list.py')
| -rw-r--r-- | lib/utils/list.py | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/lib/utils/list.py b/lib/utils/list.py new file mode 100644 index 00000000..efa3969e --- /dev/null +++ b/lib/utils/list.py @@ -0,0 +1,23 @@ +import random + + +def _uniform_rng(): + while True: + yield random.uniform(0, 1) + + +_rng = _uniform_rng() + + +def poprandom(sequence, rng=_rng): + index = int(round(next(rng) * (len(sequence) - 1))) + item = sequence[index] + + # It's O(1) to pop the last item, and O(n) to pop any other item. So we'll + # always pop the last item and put it in the slot vacated by the item we're + # popping. + last_item = sequence.pop() + if index < len(sequence): + sequence[index] = last_item + + return item |
