Cutting numbers into fixed buckets

cutting_data_in_buckets.py

"""Cutting numbers into fixed buckets

This is a solution for StackOverflow question
https://datascience.stackexchange.com/questions/45118/cutting-numbers-into-fixed-buckets
"""

from bisect import bisect
from random import sample

data = sample(range(10_000), 1_000)

breakpoints = [1, 5, 25, 50, 150, 250, 1_000, 5_000, 10_000]
buckets = {}

for i in data:
    buckets.setdefault(breakpoints[bisect(breakpoints, i)], []).append(i)