具有自定义浮动格式的pprint

时间:2017-06-04 17:07:52

标签: python 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可以实现,但我不知道如何做到这一点。

感谢。

1 个答案:

答案 0 :(得分:3)

正如您所说,您可以通过继承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)

示例:

>>> 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}}