我有一个带有元组键的嵌套字典结构。这是当我使用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
可以实现,但我不知道如何做到这一点。
感谢。
答案 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}}