具有自定义浮点格式的 pprint

我有一个带有元组键的嵌套字典结构。当我使用 pprint 漂亮地打印字典时，条目如下所示：

...
 ('A', 'B'): {'C': 0.14285714285714285,
              'D': 0.14285714285714285,
              'E': 0.14285714285714285,
              'F': 0.14285714285714285,
              'G': 0.14285714285714285,
              'H': 0.14285714285714285,
              'I': 0.14285714285714285},
...

它非常漂亮，但我想通过减少浮点数中的一些额外数字来进一步定制它。我认为可以通过子类化来实现pprint.PrettyPrint但我不知道该怎么做。

Thanks.

正如你所说，你可以通过子类化来实现这一点PrettyPrinter并覆盖format方法。请注意，输出不仅是格式化的字符串，还包括一些标志。

一旦完成，您还可以概括这一点，并将具有不同类型所需格式的字典传递到构造函数中：

class FormatPrinter(pprint.PrettyPrinter):

    def __init__(self, formats):
        super(FormatPrinter, self).__init__()
        self.formats = formats

    def format(self, obj, ctx, maxlvl, lvl):
        if type(obj) in self.formats:
            return self.formats[type(obj)] % obj, 1, 0
        return pprint.PrettyPrinter.format(self, obj, ctx, maxlvl, lvl)

Example:

>>> d = {('A', 'B'): {'C': 0.14285714285714285,
...                   'D': 0.14285714285714285,
...                   'E': 0.14285714285714285},
...       'C': 255}
...
>>> FormatPrinter({float: "%.2f", int: "%06X"}).pprint(d)
{'C': 0000FF,
 ('A', 'B'): {'C': 0.14,
              'D': 0.14,
              'E': 0.14}}