bucket.py

from typing import List

# 固定的base
def numeric(array: List[float], reverse: bool=False) -> None:
    '''
    array: 支持数值型数据，如整型与浮点型混合；不支持含有字符串类型的数据。
    reverse: 是否降序, 默认采用升序。
    '''
    if not array:
        return None
    arrmin = min(array)
    arrmax = max(array)
    length = len(array)
    capacity = int(arrmax - arrmin + 1) // length + 1
    bucket = [[] for _ in range(capacity)] # Construct buckets
    for value in array:
        pos = int(value - arrmin) // length
        bucket[capacity - pos - 1 if reverse else pos].append(value)
    array.clear()
    for index in bucket:
        if index:
            for value in sorted(index, reverse=reverse): # TimSort
                array.append(value)

# 基数越大，桶数越小
def mobase(array: List[float], base: int=5, reverse: bool=False) -> None:
    '''
    array: 支持数值型数据，如整型与浮点型混合；不支持含有字符串类型的数据。
    base: 根据需要调节, base 越小, 桶数越大。
    reverse: 是否降序, 默认采用升序。
    '''
    if not array:
        return None
    assert base > 0
    arrmin = min(array)
    arrmax = max(array)
    capacity = int(arrmax - arrmin + 1) // base + 1
    bucket = [[] for _ in range(capacity)]
    for value in array:
        pos = int(value - arrmin) // base
        bucket[capacity - pos - 1 if reverse else pos].append(value)
    array.clear()
    for index in bucket:
        if index:
            for value in sorted(index, reverse=reverse): # TimSort
                array.append(value)