Task: Bin given limits




Go: package main

import (
    "fmt"
    "sort"
)

func getBins(limits, data []int) []int {
    n := len(limits)
    bins := make([]int, n+1)
    for _, d := range data {
        index := sort.SearchInts(limits, d) 
        if index < len(limits) && d == limits[index] {
            index++
        }
        bins[index]++
    }
    return bins
}

func printBins(limits, bins []int) {
    n := len(limits)
    fmt.Printf("           < %3d = %2d\n", limits[0], bins[0])
    for i := 1; i < n; i++ {
        fmt.Printf(">= %3d and < %3d = %2d\n", limits[i-1], limits[i], bins[i])
    }
    fmt.Printf(">= %3d           = %2d\n", limits[n-1], bins[n])
    fmt.Println()
}

func main() {
    limitsList := [][]int{
        {23, 37, 43, 53, 67, 83},
        {14, 18, 249, 312, 389, 392, 513, 591, 634, 720},
    }

    dataList := [][]int{
        {
            95, 21, 94, 12, 99, 4, 70, 75, 83, 93, 52, 80, 57, 5, 53, 86, 65, 17, 92, 83, 71, 61, 54, 58, 47,
            16,  8,  9, 32, 84, 7, 87, 46, 19, 30, 37, 96, 6, 98, 40, 79, 97, 45, 64, 60, 29, 49, 36, 43, 55,
        },
        {
            445, 814, 519, 697, 700, 130, 255, 889, 481, 122, 932,  77, 323, 525, 570, 219, 367, 523, 442, 933,
            416, 589, 930, 373, 202, 253, 775,  47, 731, 685, 293, 126, 133, 450, 545, 100, 741, 583, 763, 306,
            655, 267, 248, 477, 549, 238,  62, 678,  98, 534, 622, 907, 406, 714, 184, 391, 913,  42, 560, 247,
            346, 860,  56, 138, 546,  38, 985, 948,  58, 213, 799, 319, 390, 634, 458, 945, 733, 507, 916, 123,
            345, 110, 720, 917, 313, 845, 426,   9, 457, 628, 410, 723, 354, 895, 881, 953, 677, 137, 397,  97,
            854, 740,  83, 216, 421,  94, 517, 479, 292, 963, 376, 981, 480,  39, 257, 272, 157,   5, 316, 395,
            787, 942, 456, 242, 759, 898, 576,  67, 298, 425, 894, 435, 831, 241, 989, 614, 987, 770, 384, 692,
            698, 765, 331, 487, 251, 600, 879, 342, 982, 527, 736, 795, 585,  40,  54, 901, 408, 359, 577, 237,
            605, 847, 353, 968, 832, 205, 838, 427, 876, 959, 686, 646, 835, 127, 621, 892, 443, 198, 988, 791,
            466,  23, 707, 467,  33, 670, 921, 180, 991, 396, 160, 436, 717, 918,   8, 374, 101, 684, 727, 749,
        },
    }

    for i := 0; i < len(limitsList); i++ {
        fmt.Println("Example", i+1, "\b\n")
        bins := getBins(limitsList[i], dataList[i])
        printBins(limitsList[i], bins)
    }
}




Python: from bisect import bisect_right

def bin_it(limits: list, data: list) -> list:
    "Bin data according to (ascending) limits."
    bins = [0] * (len(limits) + 1)      
    for d in data:
        bins[bisect_right(limits, d)] += 1
    return bins

def bin_print(limits: list, bins: list) -> list:
    print(f"          < {limits[0]:3} := {bins[0]:3}")
    for lo, hi, count in zip(limits, limits[1:], bins[1:]):
        print(f">= {lo:3} .. < {hi:3} := {count:3}")
    print(f">= {limits[-1]:3}          := {bins[-1]:3}")


if __name__ == "__main__":
    print("RC FIRST EXAMPLE\n")
    limits  = [23, 37, 43, 53, 67, 83]
    data = [95,21,94,12,99,4,70,75,83,93,52,80,57,5,53,86,65,17,92,83,71,61,54,58,47,
            16, 8, 9,32,84,7,87,46,19,30,37,96,6,98,40,79,97,45,64,60,29,49,36,43,55]
    bins = bin_it(limits, data)
    bin_print(limits, bins)

    print("\nRC SECOND EXAMPLE\n")
    limits = [14, 18, 249, 312, 389, 392, 513, 591, 634, 720]
    data = [445,814,519,697,700,130,255,889,481,122,932, 77,323,525,570,219,367,523,442,933,
            416,589,930,373,202,253,775, 47,731,685,293,126,133,450,545,100,741,583,763,306,
            655,267,248,477,549,238, 62,678, 98,534,622,907,406,714,184,391,913, 42,560,247,
            346,860, 56,138,546, 38,985,948, 58,213,799,319,390,634,458,945,733,507,916,123,
            345,110,720,917,313,845,426,  9,457,628,410,723,354,895,881,953,677,137,397, 97,
            854,740, 83,216,421, 94,517,479,292,963,376,981,480, 39,257,272,157,  5,316,395,
            787,942,456,242,759,898,576, 67,298,425,894,435,831,241,989,614,987,770,384,692,
            698,765,331,487,251,600,879,342,982,527,736,795,585, 40, 54,901,408,359,577,237,
            605,847,353,968,832,205,838,427,876,959,686,646,835,127,621,892,443,198,988,791,
            466, 23,707,467, 33,670,921,180,991,396,160,436,717,918,  8,374,101,684,727,749]
    bins = bin_it(limits, data)
    bin_print(limits, bins)