Python字符串格式化：实现最多两位小数且无末尾零的输出

问题描述 #

我有一些包含0个或多个小数位的浮点数，希望将它们格式化为最多保留两位小数且无末尾零的形式。即需要定义一个magic_fmt函数，满足以下示例：

>>> ls = [123, 12, 1, 1.2, 1.23, 1.234, 1.2345]
>>> [magic_fmt(n, 2) for n in ls]
['123', '12', '1', '1.2', '1.23', '1.23', '1.23']

我尝试了几种方法：

>>> [f'{n:.2e}' for n in ls] # 完全不符合预期
['1.23e+02',
 '1.20e+01',
 '1.00e+00',
 '1.20e+00',
 '1.23e+00',
 '1.23e+00',
 '1.23e+00']
>>> [f'{n:.2f}' for n in ls] # 存在末尾零
['123.00', '12.00', '1.00', '1.20', '1.23', '1.23', '1.23']
>>> [f'{n:.2g}' for n in ls] # 意外切换为科学计数法
['1.2e+02', '12', '1', '1.2', '1.2', '1.2', '1.2']
>>> [str(round(n,2)) for n in ls]
['123', '12', '1', '1.2', '1.23', '1.23', '1.23']

目前只有放弃字符串格式化迷你语言的方法可行，但这种方式存在风险，请问是否有更优的实现方案？

可行方案 #

方案1：格式化+字符串修剪 #

先通过格式化语法完成四舍五入并保留指定小数位，再移除末尾的零和多余小数点，兼顾效率和结果要求：

def magic_fmt(n, decimals=2):
    # 格式化为指定小数位的字符串
    formatted = f"{n:.{decimals}f}"
    # 移除末尾零，若只剩小数点则一并移除
    return formatted.rstrip('0').rstrip('.') if '.' in formatted else formatted

测试验证：

ls = [123, 12, 1, 1.2, 1.23, 1.234, 1.2345]
print([magic_fmt(n, 2) for n in ls])
# 输出: ['123', '12', '1', '1.2', '1.23', '1.23', '1.23']

该方法避免了round可能带来的浮点数精度异常，同时利用格式化语法完成四舍五入。

方案2：Decimal模块精确处理 #

针对对精度要求高的场景，使用decimal模块规避浮点数二进制存储的精度误差，同时自动处理末尾零：

from decimal import Decimal, ROUND_HALF_UP

def magic_fmt(n, decimals=2):
    # 转换为Decimal对象，指定四舍五入规则
    decimal_num = Decimal(str(n)).quantize(
        Decimal(f'1.{"0"*decimals}'),
        rounding=ROUND_HALF_UP
    )
    # normalize()自动移除末尾零，to_eng_string()避免科学计数法
    return decimal_num.normalize().to_eng_string()

测试验证：

ls = [123, 12, 1, 1.2, 1.23, 1.234, 1.2345]
print([magic_fmt(n, 2) for n in ls])
# 输出: ['123', '12', '1', '1.2', '1.23', '1.23', '1.23']

此方案能精准处理四舍五入，尤其适合金融、计量等对精度敏感的场景。

方案3：优化round方法 #

若偏好使用round，可通过字符串处理优化结果，同时尽量降低浮点数精度风险：

def magic_fmt(n, decimals=2):
    rounded_num = round(n, decimals)
    result_str = str(rounded_num)
    # 仅当包含小数点时，移除末尾零和多余小数点
    if '.' in result_str:
        result_str = result_str.rstrip('0').rstrip('.')
    return result_str

注意：部分浮点数因二进制存储特性，round可能出现预期外结果（如round(2.675,2)得到2.67而非2.68），此时推荐方案2。

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