如何在Python中对列表定义的自定义进制执行加减运算？

对应专业术语 #

这种操作对应的术语是混合进制数（Mixed Radix Number），也叫变进制数，指计数系统中每一位的进位基数各不相同，这里的三个列表分别对应混合进制的不同数位，各自的长度就是对应数位的基数。

实现方案（Python） #

我们可以通过自定义类结合双向映射，实现序列元素与整数的转换，进而支持加减运算：

1. 构建基础映射 #

先定义原始列表，生成所有序列元素并建立元素与整数索引的双向映射：

a = ["alpha", "beta", "gamma"]
b = ["a", "b", "c", "d"]
c = ["one", "two", "three", "four", "five"]

# 生成完整序列并创建映射
all_elements = []
for ai in a:
    for bi in b:
        for ci in c:
            all_elements.append(f"{ai}-{bi}-{ci}")

element_to_num = {elem: idx for idx, elem in enumerate(all_elements)}
num_to_element = {idx: elem for idx, elem in enumerate(all_elements)}

2. 自定义运算类 #

通过重载Python的运算符方法，让序列元素实例支持加减操作：

class CustomMixedRadix:
    def __init__(self, element):
        self.num = element_to_num[element]
    
    def __sub__(self, other):
        if isinstance(other, CustomMixedRadix):
            return self.num - other.num
        raise TypeError("仅支持与CustomMixedRadix实例相减")
    
    def __add__(self, other):
        if isinstance(other, int):
            new_num = self.num + other
            if new_num < 0 or new_num >= len(all_elements):
                raise ValueError("运算结果超出序列范围")
            return CustomMixedRadix(num_to_element[new_num])
        raise TypeError("仅支持与整数相加")
    
    def __repr__(self):
        return num_to_element[self.num]

3. 测试示例 #

按照需求测试运算：

# 创建实例
num1 = CustomMixedRadix("alpha-b-two")
num2 = CustomMixedRadix("alpha-a-four")

# 减法运算
print(num1 - num2)  # 输出: 3

# 加法运算
num = CustomMixedRadix("alpha-a-four")
print(num + 3)  # 输出: alpha-b-two

核心逻辑 #

序列元素的整数索引遵循混合进制的计算规则：索引值 = a元素的索引 × (b的长度 × c的长度) + b元素的索引 × c的长度 + c元素的索引。加减运算本质是对整数索引进行操作，再反向转换为序列元素字符串。

内容的提问来源于stack exchange，提问作者HanMah