能否为Python的f-string覆盖默认格式化器？

Reading through PEP 3101，它讨论了如何子类化string.Formatter来定义自己的格式，但使用时需要通过myformatter.format(string)。有没有办法让它直接和f-string一起工作？

例如，我想实现类似f"height = {height:.2f}"的写法，但要用我自己的浮点数格式化器来处理某些特殊情况。

其实f-string本身是直接调用目标对象的__format__方法来完成格式化的，没办法直接全局替换它使用的默认格式化器，但我们有几种实用的替代方案来实现自定义格式化效果：

• 给自定义类型实现__format__方法
  如果是你自己定义的类，直接在类里实现__format__方法，就能让f-string自动使用你的自定义逻辑。要是需要处理原生float的特殊情况，可以把float包装成自定义类：

  class MyFloat:
      def __init__(self, value):
          self.value = value
      def __format__(self, format_spec):
          # 这里写你的特殊处理逻辑，比如处理无穷大或者NaN
          if self.value == float('inf'):
              return "INF"
          # 其他情况可以复用默认格式化逻辑，再加上自定义规则
          return f"{self.value:.{format_spec}f}" if format_spec else str(self.value)
  
  height = MyFloat(1.2345)
  print(f"height = {height:.2f}")  # 会按照自定义逻辑输出
  

• 临时替换原生类型的__format__方法（谨慎使用）
  如果你不想包装类型，可以临时替换float的原生__format__方法，但要注意这是全局修改，可能影响其他代码，用完记得恢复：

  # 先保存原生方法
  original_float_format = float.__format__
  
  def custom_float_format(self, format_spec):
      # 比如给负数加上终端红色高亮
      if self < 0:
          return f"\033[91m{original_float_format(self, format_spec)}\033[0m"
      return original_float_format(self, format_spec)
  
  # 替换方法
  float.__format__ = custom_float_format
  
  height = -1.2345
  print(f"height = {height:.2f}")  # 负数会显示红色
  # 恢复原生方法，避免影响其他代码
  float.__format__ = original_float_format
  

• 用辅助函数在f-string中转换
  要是不想修改类型或全局方法，也可以写个辅助函数，在f-string里直接调用它来完成自定义格式化：

  def format_my_float(value, format_spec):
      # 特殊处理0的显示
      if value == 0:
          return "0.00"
      return f"{value:.{format_spec}f}"
  
  height = 0.0
  print(f"height = {format_my_float(height, '2f')}")
  

简单来说，f-string没有直接替换全局格式化器的入口，但通过上述几种方式，完全可以实现你想要的自定义格式化效果。

备注：内容来源于stack exchange，提问作者Edward Falk