Python向下截断函数优化：如何减少条件判断并支持t位泛化？

原函数存在未定义变量、冗余条件分支、未处理负数等问题，且难以直接泛化到任意位数。下面是一个简化实现，无需过多条件判断，同时支持自定义截断位数t：

def trunc_to_t_digits(num, t=3):
    # 分离符号与数字部分
    num_str = str(num)
    sign = '-' if num_str.startswith('-') else ''
    core_num = num_str.lstrip('-')
    
    # 处理全零情况
    if all(c in ('0', '.') for c in core_num):
        return 0.0
    
    # 找到第一个非零数字的位置（跳过小数点）
    first_non_zero = next(i for i, c in enumerate(core_num) if c not in ('0', '.'))
    
    # 提取所有数字字符（移除小数点），方便统一处理
    digits_only = [c for c in core_num if c != '.']
    # 保留第一个非零开始的t位数字，其余补0
    truncated_digits = digits_only[:first_non_zero + t] + ['0'] * max(0, len(digits_only) - (first_non_zero + t))
    
    # 重新插入小数点（如果原数有小数点）
    if '.' in core_num:
        decimal_pos = core_num.index('.')
        # 确保小数点位置不超出截断后的数字长度
        if decimal_pos <= len(truncated_digits):
            truncated_digits.insert(decimal_pos, '.')
        else:
            truncated_digits.append('.')
    
    # 拼接结果，清理末尾多余的0和小数点，加上符号
    result_str = sign + ''.join(truncated_digits).rstrip('0').rstrip('.')
    # 处理极端情况（理论上不会触发，因为全零已提前处理）
    result_str = result_str or '0'
    
    return float(result_str)

关键思路说明 #

• 统一处理正负号：先分离符号，最后再拼接回去，避免符号干扰数字处理逻辑。
• 数字与小数点解耦：先把所有数字提取出来处理截断，再重新插入小数点，无需区分有无小数点的分支。
• 泛化截断逻辑：通过参数t控制保留的位数，只需修改t的值就能支持任意位数的截断需求。
• 自动清理冗余字符：最后通过rstrip去掉末尾多余的0和小数点，保证输出格式整洁。

测试示例 #

print(trunc_to_t_digits(1234.56))       # 输出 1230.0
print(trunc_to_t_digits(0.001234))      # 输出 0.00123
print(trunc_to_t_digits(-9876.54, 4))   # 输出 -9876.0
print(trunc_to_t_digits(0))             # 输出 0.0

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