基数排序python语法

def radix_sort(arr):
    # 获取数组中最大值的位数
    max_value = max(arr)
    max_digits = len(str(max_value))
    
    # 初始化桶
    buckets = [[] for _ in range(10)]
    
    # 进行每一位的排序
    for digit in range(max_digits):
        # 将元素放入对应的桶中
        for num in arr:
            index = (num // 10 ** digit) % 10
            buckets[index].append(num)
        
        # 将桶中的元素按顺序放回原数组
        arr = [num for bucket in buckets for num in bucket]
        
        # 清空桶
        buckets = [[] for _ in range(10)]
    
    return arr

# 示例用法
arr = [170, 45, 75, 90, 802, 24, 2, 66]
sorted_arr = radix_sort(arr)
print(sorted_arr)