我正在使用一个可以打印许多浮点数的 Python 脚本。 在每个
print
命令中,我在打印之前使用 round(x, 3)
将打印的数字 x
四舍五入到小数点后 3 位。 这是一个简化的示例:
x1 = 1.61803398875
print('x1 = ' + str(round(x1, 3))) # x1 = 1.618
x2 = x1**2
print('x2 = ' + str(round(x2, 3))) # x2 = 2.618
x3 = 1/(x2-1)
print('x3 = ' + str(round(x3, 3))) # x3 = 0.618
实际代码包含数百个这样的打印命令。 有没有更有效的方法来执行此脚本的舍入部分? 例如,是否有一个总体命令可以将所有打印的浮点数四舍五入到给定的精度,就像
numpy
和 pandas
变量一样?
重要的是,我正在寻找一种解决方案,不涉及移动或组合
print
命令或在打印之前将打印的变量转换为 numpy
和 pandas
变量。
使用 f 字符串是指定数字字符串表示形式舍入的更简单方法:
x1 = 1.61803398875
print(f'x1 = {x1:.3f}') # x1 = 1.618
x2 = x1**2
print(f'x2 = {x2:.3f}') # x2 = 2.618
x3 = 1/(x2-1)
print(f'x3 = {x3:.3f}') # x3 = 0.618
但是你仍然需要每次重复
.3f
部分。
如果你想尽可能干燥,你可以写一个包装器:
def print_var(var: str, val: float) -> None:
print(f'{var} = {val:.3f}')
x1 = 1.61803398875
print_var('x1', x1) # x1 = 1.618
x2 = x1**2
print_var('x2', x2) # x2 = 2.618
x3 = 1/(x2-1)
print_var('x3', x3) # x3 = 0.618
或者你可以重新定义
print
(我个人不会推荐这个):
import builtins
def print(*args, **kwargs):
rounded_args = [
f'{arg:.3f}' if isinstance(arg, float) else arg
for arg in args
]
builtins.print(*rounded_args, **kwargs)
x1 = 1.61803398875
print('x1 = ', x1) # x1 = 1.618
x2 = x1**2
print('x2 = ', x2) # x2 = 2.618
x3 = 1/(x2-1)
print('x3 = ', x3) # x3 = 0.618