cp_library/alg/iter/sort/sort_buckets_fn.py

• View this file on GitHub
• Last update: 2025-06-20 03:24:59+09:00

Code

import cp_library.__header__
from itertools import groupby
from operator import itemgetter
import cp_library.alg.__header__
import cp_library.alg.iter.__header__
import cp_library.alg.iter.sort.__header__

def sort_buckets(A, N, key=0):
    if isinstance(key,int): key = itemgetter(key)
    B = [[] for _ in range(N)]; A.sort(key=key)
    for k, g in groupby(A, key=key): B[k] = list(g)
    return B
    

'''
╺━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━╸
             https://kobejean.github.io/cp-library               
'''
from itertools import groupby
from operator import itemgetter




def sort_buckets(A, N, key=0):
    if isinstance(key,int): key = itemgetter(key)
    B = [[] for _ in range(N)]; A.sort(key=key)
    for k, g in groupby(A, key=key): B[k] = list(g)
    return B
    

Back to top page