如何用Python打印带数字上下标的字符串（函数表达式、化学方程式等）

打印上下标可以采用str对象的maketrans(str1, str2)方法。该方法用于创建字符映射转换表，有两个参数，参数str1存入要转换的字符，参数str存入相应的转换后的字符，例如大小写字母的转换：

upper_letter = 'ABCDEFGHIJKLMNOPQRSTUVWXYZ'
lower_letter = 'abcdefghijklmnopqrstuvwxyz'
# 创建字符映射表
upper_to_lower = str.maketrans(upper_letter, lower_letter)
print('PYTHON3.8'.translate(upper_to_lower))
# 输出：python3.8

输出结果显示，映射表中包含的大写字母（PYTHON）都转换成了小写（python），映射表中不涉及的字符（3.8）不发生变化。

对于打印数字上下标的需求，只需创建好普通样式和上标样式的数字字符映射表，就能实现转换。比如打印方程式需要上标格式的数字：

# 定义上标数字的映射表
sup_map = str.maketrans('0123456789', '⁰¹²³⁴⁵⁶⁷⁸⁹')
formula = 'y=x2'
print(formula.translate(sup_map))
# 输出：y=x²

再比如打印化学分子式需要下标格式的数字：

# 定义下标数字的映射表
sub_map = str.maketrans('0123456789', '₀₁₂₃₄₅₆₇₈₉')
formula = 'H2O'
print(formula.translate(sub_map))
# 输出：H₂O

更复杂的输出可以结合正则表达式，比如下面的代码输出的是一元三次函数：

# 创建字符映射表
sup_map = str.maketrans('0123456789', '⁰¹²³⁴⁵⁶⁷⁸⁹')
# 要转换的字符串
formula = 'y=x3+2x2+3x+4'
# 匹配出要转换的表示次幂的字符
results = re.findall(r'x\d\+', formula)
# 依次替换成上标的格式
for s in results:
    # s[:-1]的目的是让结尾的加号(+)不参与替换操作，因为“+”与通配符有冲突
    formula = re.sub(s[:-1], s[:-1].translate(sup_map), formula)
print(formula)  # 输出：y=x³+2x²+3x+4

参考资料：https://codeigo.com/python/printing-subscript-and-superscript