Python: Checking to which bin a value belongs

by

Probably too late, but for future reference, numpy has a function that does just that:

http://docs.scipy.org/doc/numpy/reference/generated/numpy.digitize.html

>>> my_list = [3,2,56,4,32,4,7,88,4,3,4]
>>> bins = [0,20,40,60,80,100]
>>> np.digitize(my_list,bins)
array([1, 1, 3, 1, 2, 1, 1, 5, 1, 1, 1])

The result is an array of indexes corresponding to the bin from bins that each element from my_list belongs too.
Note that the function will also bin values that fall outside of your first and last bin edges:

>>> my_list = [-5,200]
>>> np.digitize(my_list,bins)
array([0, 6])

And Pandas has something like it too:

http://pandas.pydata.org/pandas-docs/dev/basics.html#discretization-and-quantiling

>>> pd.cut(my_list, bins)
Categorical: 
array(['(0, 20]', '(0, 20]', '(40, 60]', '(0, 20]', '(20, 40]', '(0, 20]',
       '(0, 20]', '(80, 100]', '(0, 20]', '(0, 20]', '(0, 20]'], dtype=object)
Levels (5): Index(['(0, 20]', '(20, 40]', '(40, 60]', '(60, 80]',
                   '(80, 100]'], dtype=object)

More Related Contents:

  1. Search list for number in range
  2. What happens when start value < end value with range() cmd in Python?
  3. How to use a decimal range() step value?
  4. Identify groups of continuous numbers in a list
  5. What is the difference between range and xrange functions in Python 2.X?
  6. Why does range(start, end) not include end?
  7. Should you always favor xrange() over range()?
  8. Pandas: convert categories to numbers
  9. range() for floats
  10. Is it possible to implement a Python for range loop without an iterator variable?
  11. `xrange(2**100)` -> OverflowError: long int too large to convert to int
  12. NameError: global name ‘xrange’ is not defined in Python 3
  13. converting a list of integers into range in python
  14. How to check if the current time is in range in python?
  15. How to loop backwards in python? [duplicate]
  16. Python range() and zip() object type
  17. How to find range overlap in python?
  18. How does the Python’s range function work?
  19. 2D and 3D Scatter Histograms from arrays in Python
  20. Print a list in reverse order with range()?
  21. Range as dictionary key in Python
  22. Is there a need for range(len(a))?
  23. Python 3 range Vs Python 2 range
  24. Python IndexError: list index out of range
  25. Is it possible to make a `for` loop without an iterator variable? (How can I make make code loop a set number of times?)
  26. Python: Mapping from intervals to values
  27. Skip over a value in the range function in python
  28. Does range() really create lists?
  29. Limiting user input to a range in Python
  30. Strange result when removing item from a list while iterating over it in Python

Leave a Comment