Sampling an infinite stream with reservoir sampling

reservoir-sampling.py

#!/usr/bin/python

import random

shuf = random.sample(range(100000), 1000)

# Keep bag of k=10 elements
# Prefill the bag with first shuf items
bag = shuf[:10]

for i in shuf[10:]:
  k = random.randint(0, 9)

  # Update bag in 1/k chance
  if k == 1:
    # Randomly kick out old bag item
    indx = random.randint(0, 9)
    bag[indx] = i

print bag