Random weighted choice

by

Generate a Cumulative Distribution Function for each ID1 thus:

cdfs = defaultdict()
for id1,id2,val in d:
    prevtotal = cdfs[id1][-1][0]
    newtotal = prevtotal + val
    cdfs[id1].append( (newtotal,id2) )

So you will have

cdfs = { 701 : [ (0.2,1), (0.5,2), (1.0,3) ], 
         702 : [ (0.2,1), (0.5,2) ],
         703 : [ (0.5,3) ] }

Then generate a random number and search for it in the list.

def func(id1):
    max = cdfs[id1][-1][0]
    rand = random.random()*max
    for upper,id2 in cdfs[id1]:
        if upper>rand:
            return id2
    return None

More Related Contents:

  1. int has no len error Help me please? [closed]
  2. Why do people write #!/usr/bin/env python on the first line of a Python script?
  3. Request UAC elevation from within a Python script?
  4. Using headers with the Python requests library’s get method
  5. Why are trailing commas allowed in a list?
  6. Replacing Numpy elements if condition is met
  7. Can existing virtualenv be upgraded gracefully?
  8. Add Variables to Tuple
  9. Underscore vs Double underscore with variables and methods [duplicate]
  10. No module named when using PyInstaller
  11. How to update SQLAlchemy row entry?
  12. Selenium Webdriver in Python – files download directory change in Chrome preferences
  13. @csrf_exempt does not work on generic view based class
  14. How to transform a tuple to a string of values without comma and parentheses
  15. How can I open the interactive matplotlib window in IPython notebook?
  16. How to get only the last part of a path in Python?
  17. Adding values for missing data combinations in Pandas
  18. Can python load definitions from a C header file?
  19. Python – re.findall returns unwanted result
  20. Python get current time in right timezone [duplicate]
  21. Why is pandas.to_datetime slow for non standard time format such as ‘2014/12/31’
  22. How do convert unicode escape sequences to unicode characters in a python string
  23. How can I print multiple things (fixed text and/or variable values) on the same line, all at once?
  24. Why are integers immutable in Python?
  25. How indexing works in Pandas?
  26. Skip over a value in the range function in python
  27. Python 2.7: Print thread safe
  28. Why is lxml.etree.iterparse() eating up all my memory?
  29. Get total values_count from a dataframe with Python Pandas
  30. Copy complete virtualenv to another pc

Leave a Comment